From 84e76292e0977c3787733d88cd8ad4a6d258257f Mon Sep 17 00:00:00 2001
From: YunLiu <55491388+KumoLiu@users.noreply.github.com>
Date: Fri, 1 Mar 2024 10:47:37 +0800
Subject: [PATCH] Revert "remove self._probs in initialize"

This reverts commit 013a8bfef3a63bb110ffcb61bbf227a33777d180.
---
 bundle/python_bundle_workflow/scripts/inference.py | 2 +-
 bundle/python_bundle_workflow/scripts/train.py     | 1 +
 2 files changed, 2 insertions(+), 1 deletion(-)

diff --git a/bundle/python_bundle_workflow/scripts/inference.py b/bundle/python_bundle_workflow/scripts/inference.py
index 06ce1aff5f..24810a0dfb 100644
--- 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
index 16fa85b324..ccd409e969 100644
--- 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()