Use data_dir instead of flags.FLAGS.data_dir in data_preprocessing.py.

shizhiw · shizhiw · commit cb948dafc405 · 2018-10-11T16:33:49.000-07:00
diff --git a/official/recommendation/data_preprocessing.py b/official/recommendation/data_preprocessing.py
@@ -450,7 +450,7 @@ def instantiate_pipeline(dataset, data_dir, batch_size, eval_batch_size,
   }
   if ncf_dataset.deterministic:
     flags_["seed"] = stat_utils.random_int32()
-  tf.gfile.MakeDirs(flags.FLAGS.data_dir)
+  tf.gfile.MakeDirs(data_dir)
   # We write to a temp file then atomically rename it to the final file,
   # because writing directly to the final file can cause the data generation
   # async process to read a partially written JSON file.

Original file line number	Diff line number	Diff line change
`@@ -450,7 +450,7 @@ def instantiate_pipeline(dataset, data_dir, batch_size, eval_batch_size,`
`450`	`450`	`}`
`451`	`451`	`if ncf_dataset.deterministic:`
`452`	`452`	`flags_["seed"] = stat_utils.random_int32()`
`453`		`- tf.gfile.MakeDirs(flags.FLAGS.data_dir)`
	`453`	`+ tf.gfile.MakeDirs(data_dir)`
`454`	`454`	`# We write to a temp file then atomically rename it to the final file,`
`455`	`455`	`# because writing directly to the final file can cause the data generation`
`456`	`456`	`# async process to read a partially written JSON file.`