Revert "remove self._set_prop"

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

Hide file tree

Showing 2 changed files with 10 additions and 2 deletions.
diff --git a/bundle/python_bundle_workflow/scripts/inference.py b/bundle/python_bundle_workflow/scripts/inference.py
@@ -63,6 +63,7 @@ def __init__(self, dataset_dir: str = "."):
             nib.save(n, os.path.join(dataset_dir, f"seg{i:d}.nii.gz"))
 
         self._props = {}
+        self._set_props = {}
         self.dataset_dir = dataset_dir
 
     def initialize(self):
@@ -75,7 +76,7 @@ def finalize(self):
         pass
 
     def _set_property(self, name, property, value):
-        self._props[name] = value
+        self._set_props[name] = value
 
     def _get_property(self, name, property):
         """
@@ -89,6 +90,9 @@ def _get_property(self, name, property):
         value = None
         if name in self._props:
             value = self._props[name]
+        elif name in self._set_props:
+            value = self._set_props[name]
+            self._props[name] = value
         else:
             try:
                 value = getattr(self, f"get_{name}")()

diff --git a/bundle/python_bundle_workflow/scripts/train.py b/bundle/python_bundle_workflow/scripts/train.py
@@ -72,6 +72,7 @@ def __init__(self, dataset_dir: str = "."):
 
         # define buckets to store the generated properties and set properties
         self._props = {}
+        self._set_props = {}
         self.dataset_dir = dataset_dir
 
         # besides the predefined properties, this bundle workflow can also provide `network`, `loss`, `optimizer`
@@ -89,7 +90,7 @@ def finalize(self):
         set_determinism(None)
 
     def _set_property(self, name, property, value):
-        self._props[name] = value
+        self._set_props[name] = value
 
     def _get_property(self, name, property):
         """
@@ -103,6 +104,9 @@ def _get_property(self, name, property):
         value = None
         if name in self._props:
             value = self._props[name]
+        elif name in self._set_props:
+            value = self._set_props[name]
+            self._props[name] = value
         else:
             try:
                 value = getattr(self, f"get_{name}")()