diff --git a/tf_adapter/python/npu_bridge/estimator/npu/npu_estimator.py b/tf_adapter/python/npu_bridge/estimator/npu/npu_estimator.py index a06b9a38c93f0ec3afb1daafe708e26d775d845a..44f1b8ab158673569c579a7797914f3915bb2a9f 100644 --- a/tf_adapter/python/npu_bridge/estimator/npu/npu_estimator.py +++ b/tf_adapter/python/npu_bridge/estimator/npu/npu_estimator.py @@ -377,7 +377,9 @@ class NPUEstimator(estimator_lib.Estimator): config = self.__load_graph_optimizers(config) # Init npu system: get task and device info from configuration file. - self.__load_job_info(job_start_file) + if !(self.__load_job_info(job_start_file)): + raise ValueError( + '`config` must be provided with type `NPUConfigs`') # Check modie dir in NPUEstimator and NPURunConfig model_dir = self.__check_model_dir(model_dir, config)