import BuildConfiguration

zenml-io · Dec 15, 2023 · dd7b413 · dd7b413
1 parent 4cfac4c
commit dd7b413
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/src/zenml/integrations/spark/step_operators/kubernetes_step_operator.py b/src/zenml/integrations/spark/step_operators/kubernetes_step_operator.py
@@ -118,6 +118,8 @@ def get_docker_builds(
         Returns:
             The required Docker builds.
         """
+        from zenml.config.build_configuration import BuildConfiguration
+
         builds = []
         extra_files = {ENTRYPOINT_NAME: LOCAL_ENTRYPOINT}
         for step_name, step in deployment.step_configurations.items():