gourisankar85 commited on
Commit
5f00ea5
·
verified ·
1 Parent(s): 9b2c6f2

Update data/load_dataset.py

Browse files
Files changed (1) hide show
  1. 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: