update

Signed-off-by: Future-Outlier <eric901201@gmail.com>
flyteorg · Jan 8, 2025 · cf83e06 · cf83e06
1 parent be02f9f
commit cf83e06
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 5 deletions.
diff --git a/flytekit/core/base_task.py b/flytekit/core/base_task.py
@@ -723,10 +723,10 @@ def dispatch_execute(
           may be none
         * ``DynamicJobSpec`` is returned when a dynamic workflow is executed
         """
-        if not self.disable_deck:
+        if not self.disable_deck and ctx.user_space_params is not None:
             ctx.user_space_params._enable_deck = True  # type: ignore
-        if DeckField.TIMELINE.value in self.deck_fields and ctx.user_space_params is not None and not self.disable_deck:
-            ctx.user_space_params.decks.append(ctx.user_space_params.timeline_deck)
+            if DeckField.TIMELINE.value in self.deck_fields:
+                ctx.user_space_params.decks.append(ctx.user_space_params.timeline_deck)
         # Invoked before the task is executed
         new_user_params = self.pre_execute(ctx.user_space_params)
 

diff --git a/flytekit/core/constants.py b/flytekit/core/constants.py
@@ -38,5 +38,3 @@
 CACHE_KEY_METADATA = "cache-key-metadata"
 
 SERIALIZATION_FORMAT = "serialization-format"
-
-FLYTE_ENABLE_DECK = "FLYTE_ENABLE_DECK"
Original file line number	Diff line number	Diff line change
Expand Up		@@ -38,5 +38,3 @@
		CACHE_KEY_METADATA = "cache-key-metadata"

		SERIALIZATION_FORMAT = "serialization-format"

		FLYTE_ENABLE_DECK = "FLYTE_ENABLE_DECK"