Update data/load_dataset.py
Browse files- data/load_dataset.py +1 -1
data/load_dataset.py
CHANGED
@@ -10,7 +10,7 @@ def load_data(data_set_name):
|
|
10 |
dataset_file = os.path.join(local_path, f"{data_set_name}_test.pkl")
|
11 |
|
12 |
if os.path.exists(dataset_file):
|
13 |
-
logging.info("Loading dataset {data_set_name} from local storage")
|
14 |
with open(dataset_file, "rb") as f:
|
15 |
dataset = pickle.load(f)
|
16 |
else:
|
|
|
10 |
dataset_file = os.path.join(local_path, f"{data_set_name}_test.pkl")
|
11 |
|
12 |
if os.path.exists(dataset_file):
|
13 |
+
logging.info(f"Loading dataset {data_set_name} from local storage. File location {dataset_file}")
|
14 |
with open(dataset_file, "rb") as f:
|
15 |
dataset = pickle.load(f)
|
16 |
else:
|