fix import

Signed-off-by: Peeter Piegaze <[email protected]>
flyteorg · Jul 4, 2024 · 52d3723 · 52d3723
1 parent 92f1853
commit 52d3723
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/basic-union-template/{{cookiecutter.project_name}}/workflows/example.py b/basic-union-template/{{cookiecutter.project_name}}/workflows/example.py
@@ -1,6 +1,6 @@
 """Basic Union workflow template."""
 
-from union import task, workflow, ImageSpec
+from flytekit import task, workflow, ImageSpec
 
 # ImageSpec defines the container image used for the Kubernetes pods that run the tasks in Union.
 image_spec = ImageSpec(