Update for v0.1.5

nota-github · Sep 10, 2024 · d81020d · d81020d
1 parent e4023a8
commit d81020d
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/netspresso_inference_package/inference/inference_service.py b/netspresso_inference_package/inference/inference_service.py
@@ -17,6 +17,7 @@ def __init__(self, model_file_path:str, num_threads:int=1):
         self.model_file_path = model_file_path
         self.model_obj, self.inputs, self.outputs = self.set_model_obj(model_file_path, num_threads)
         self.result_save_path = make_temp_dir()
+        self.dataset_file_path = None
 
     def set_model_obj(self, model_file_path:str, num_threads:int):
         suffix = Path(model_file_path).suffix
@@ -60,6 +61,7 @@ def postprocess(self, inference_results:Dict[Any, np.ndarray]):
         self.result_file_path = result_file_path
 
     def run(self, dataset_file_path):
+        self.dataset_file_path = dataset_file_path
         inference_results = self.inference(dataset_file_path)
         logger.info(f"Inference success: {self.model_file_path}")
         self.postprocess(inference_results)

diff --git a/setup.py b/setup.py
@@ -5,7 +5,7 @@
 
 setup(
     name="netspresso_inference_package",
-    version="0.1.4",
+    version="0.1.5",
     author="NetsPresso",
     author_email="[email protected]",
     description="Inference module.",