Revert "remove self._probs in initialize"

This reverts commit 013a8bf.
KumoLiu · Mar 1, 2024 · 84e7629 · 84e7629
1 parent cf74488
commit 84e7629
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/bundle/python_bundle_workflow/scripts/inference.py b/bundle/python_bundle_workflow/scripts/inference.py
@@ -67,7 +67,7 @@ def __init__(self, dataset_dir: str = "."):
         self.dataset_dir = dataset_dir
 
     def initialize(self):
-        pass
+        self.props = {}
 
     def run(self):
         self.evaluator.run()

diff --git a/bundle/python_bundle_workflow/scripts/train.py b/bundle/python_bundle_workflow/scripts/train.py
@@ -82,6 +82,7 @@ def __init__(self, dataset_dir: str = "."):
 
     def initialize(self):
         set_determinism(0)
+        self.props = {}
 
     def run(self):
         self.trainer.run()