def _flags_data_(self, main_mod, model_paths, opdef):
try:
sys_path, mod_path = python_util.find_module(main_mod, model_paths)
except ImportError as e:
self.log.warning("cannot import flags from %s: %s", main_mod, e)
return {}
else:
return self.flags_data_for_path(mod_path, sys_path, opdef)
def _flags_data_(self, main_spec, model_paths, flags_dest):
try:
sys_path, mod_path = python_util.find_module(main_spec, model_paths)
except ImportError as e:
self.log.warning("cannot import flags from %s: %s", main_spec, e)
return {}
else:
package = self._main_spec_package(main_spec)
return self._flags_data_for_path(mod_path, package, sys_path, flags_dest)
def _flags_data_(self, main_mod, base_args, model_paths, flags_dest):
try:
sys_path, mod_path = python_util.find_module(main_mod, model_paths)
except ImportError as e:
if os.getenv("NO_WARN_FLAGS_IMPORT") != "1":
self.log.warning("cannot import flags from %s: %s", main_mod,
e)
return {}
else:
package = self._main_spec_package(main_mod)
return self._flags_data_for_path(mod_path, package, sys_path,
base_args, flags_dest)
def python_script_opdef_loaded(self, opdef):
if opdef.output_scalars is not None:
return
assert opdef.main, opdef
main_mod = op_util.split_cmd(opdef.main)[0]
model_paths = op_util.opdef_model_paths(opdef)
try:
_path, mod_path = python_util.find_module(main_mod, model_paths)
except ImportError:
return
script = python_util.Script(mod_path)
if self.is_keras_script(script):
if opdef.output_scalars is None:
opdef.output_scalars = KERAS_OUTPUT_SCALARS
