Merge pull request IntelLabs#181 from NervanaSystems/sharath/updated_…

…sanitize_path fix for updated sanitize_path
tnmurthy · May 10, 2018 · f90246a · f90246a
2 parents 9b867fa + f1d6fff
commit f90246a
Show file tree

Hide file tree

Showing 4 changed files with 7 additions and 7 deletions.
diff --git a/core_models/memn2n_dialogue/interactive.py b/core_models/memn2n_dialogue/interactive.py
@@ -104,9 +104,9 @@
 args = parser.parse_args()
 
 # Sanitize inputs
-model_file = sanitize_path(args.model_file)
+model_file = sanitize_path(args.model_file,prefix='')
 assert model_file.endswith('.npz')
-data_dir = sanitize_path(args.data_dir)
+data_dir = sanitize_path(args.data_dir,prefix='')
 
 babi = BABI_Dialog(
  path=data_dir,

diff --git a/core_models/memn2n_dialogue/train_model.py b/core_models/memn2n_dialogue/train_model.py
@@ -143,9 +143,9 @@
 args = parser.parse_args()
 
 # Sanitize inputs
-log_file = sanitize_path(args.log_file)
-weights_save_path = sanitize_path(args.weights_save_path)
-data_dir = sanitize_path(args.data_dir)
+log_file = sanitize_path(args.log_file,prefix='')
+weights_save_path = sanitize_path(args.weights_save_path,prefix='')
+data_dir = sanitize_path(args.data_dir,prefix='')
 assert weights_save_path.endswith('.npz')
 assert log_file.endswith('.txt')
 

diff --git a/core_models/reading_comprehension/ngraph_implementation/prepare_data.py b/core_models/reading_comprehension/ngraph_implementation/prepare_data.py
@@ -218,7 +218,7 @@ def get_ids_list(data_lists, vocab):
  print("Please enter a valid data path")
  exit()
 
- data_path = sanitize_path(args.data_path)
+ data_path = sanitize_path(args.data_path,prefix='')
 
  # Load Train and Dev Data
  train_filename = os.path.join(data_path + "/train-v1.1.json")

diff --git a/core_models/reading_comprehension/ngraph_implementation/train.py b/core_models/reading_comprehension/ngraph_implementation/train.py
@@ -97,7 +97,7 @@
  print("Please enter a valid data path")
  exit()
 
-path_gen = sanitize_path(args.data_path)
+path_gen = sanitize_path(args.data_path,prefix='')
 
 file_name_dict={}
 file_name_dict['train_para_ids']='/train.ids.context'