fix test

Signed-off-by: Kevin Su <pingsutw@apache.org>
flyteorg · Sep 25, 2024 · 6b16677 · 6b16677
1 parent 5e4076d
commit 6b16677
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/plugins/flytekit-spark/flytekitplugins/spark/task.py b/plugins/flytekit-spark/flytekitplugins/spark/task.py
@@ -196,7 +196,9 @@ def pre_execute(self, user_params: ExecutionParameters) -> ExecutionParameters:
         self.sess = sess_builder.getOrCreate()
 
         if (
-            ctx.serialization_settings.fast_serialization_settings.enabled
+            ctx.serialization_settings
+            and ctx.serialization_settings.fast_serialization_settings
+            and ctx.serialization_settings.fast_serialization_settings.enabled
             and ctx.execution_state
             and ctx.execution_state.mode == ExecutionState.Mode.TASK_EXECUTION
         ):