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