fix task: None problem while rendering

state-alchemists · Jan 8, 2024 · 57bbdc1 · 57bbdc1
1 parent 32f019a
commit 57bbdc1
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/pyproject.toml b/pyproject.toml
@@ -4,7 +4,7 @@ build-backend = "flit_core.buildapi"
 
 [project]
 name = "zrb"
-version = "0.5.0"
+version = "0.5.1"
 authors = [
   { name="Go Frendi Gunawan", email="gofrendiasgard@gmail.com" },
 ]

diff --git a/src/zrb/task/base_task/base_task.py b/src/zrb/task/base_task/base_task.py
@@ -430,6 +430,8 @@ async def _set_local_keyval(
         if self.__is_keyval_set:
             return True
         self.__is_keyval_set = True
+        # set task for rendering
+        self._set_task(self)
         # Set input_map for rendering
         self.log_info('Set input map')
         for task_input in self._get_combined_inputs():
@@ -452,8 +454,6 @@ async def _set_local_keyval(
         self.log_debug(
             'Env map:\n' + map_to_str(self.get_env_map(), item_prefix='  ')
         )
-        # set task
-        self._set_task(self)
 
     def __repr__(self) -> str:
         cls_name = self.__class__.__name__