change test file

eda/esmm/Model_pipline/train.py

@@ -12,6 +12,8 @@ from datetime import date, timedelta
 import tensorflow as tf
 import subprocess
 import time
+import glob
+import random
 
 #################### CMD Arguments ####################
 FLAGS = tf.app.flags.FLAGS
@@ -297,8 +299,7 @@ def main(_):
         FLAGS.dt_dir = (date.today() + timedelta(-1)).strftime('%Y%m%d')
     FLAGS.model_dir = FLAGS.model_dir + FLAGS.dt_dir
     #FLAGS.data_dir  = FLAGS.data_dir + FLAGS.dt_dir
-    import glob
-    import random
+
     tr_files = glob.glob("/home/gmuser/test/part*")
     random.shuffle(tr_files)
     print("tr_files:", tr_files)
@@ -369,6 +370,7 @@ def get_filename(dir_in):
 if __name__ == "__main__":
     b = time.time()
     a = "export CLASSPATH='$(hadoop classpath --glob)'"
+    print(a)
     os.system(a)
     path = "hdfs://172.16.32.4:8020/strategy/esmm/"
     tf.logging.set_verbosity(tf.logging.INFO)