丁云鹏 1 год назад
Родитель
Сommit
c8d85f9d6a
1 измененных файлов с 1 добавлено и 1 удалено
  1. 1 1
      recommend-model-produce/src/main/python/tools/static_ps_trainer_v2.py

+ 1 - 1
recommend-model-produce/src/main/python/tools/static_ps_trainer_v2.py

@@ -339,7 +339,7 @@ if __name__ == "__main__":
     read_env_flags = [
     read_env_flags = [
         key[len(flag_prefix) :]
         key[len(flag_prefix) :]
         for key in core.globals().keys()
         for key in core.globals().keys()
-        if key.startswith(flag_prefix)
+        if key.startswith("FLAGS_")
     ]
     ]
 
 
     def remove_flag_if_exists(name):
     def remove_flag_if_exists(name):