From 330de73d3920810a7ac26a5517ba9f28bf05e0c4 Mon Sep 17 00:00:00 2001 From: sadrasabouri Date: Fri, 22 Oct 2021 13:55:03 +0330 Subject: [PATCH] edit : load_data edite. --- samila/functions.py | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) 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)