diff --git a/samila/functions.py b/samila/functions.py index 916acd8..5196c84 100644 --- a/samila/functions.py +++ b/samila/functions.py @@ -231,8 +231,9 @@ def load_data(data): :return: (data1, data2) """ if isinstance(data, io.IOBase): - data = json.load(data) - if 'data1' not in data or 'data2' not in data: + try: + data = json.load(data) + return data['data1'], data['data2'] + except Exception as e: raise samilaDataError(DATA_PARSING_ERROR) - return data['data1'], data['data2'] raise samilaDataError(DATA_TYPE_ERROR)