Soften kfp version requirement, modify one import so that it works us…

…ing multiple kfp versions.
getindata · Jul 24, 2024 · 521ca3d · 521ca3d
1 parent 9c4792c
commit 521ca3d
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git a/kedro_vertexai/generator.py b/kedro_vertexai/generator.py
@@ -8,7 +8,7 @@
 
 from kedro.framework.context import KedroContext
 from kfp import dsl
-from kfp.dsl.pipeline_task import PipelineTask
+from kfp.dsl import PipelineTask
 
 from kedro_vertexai.config import (
     KedroVertexAIRunnerConfig,

diff --git a/poetry.lock b/poetry.lock
diff --git a/pyproject.toml b/pyproject.toml
@@ -35,7 +35,7 @@ known_third_party = ["click","google","kedro","kfp","kubernetes","tabulate", "py
 python = ">=3.8,<3.11" # weird error OverrideNeeded when bumping up. Could be related to old versions with kfp
 kedro = ">=0.19.0,<0.20"
 click = ">=8.0.4"
-kfp = "2.8.0"
+kfp = ">=2.0.0,<3.0"
 tabulate = ">=0.8.7"
 semver = ">=2.10,<4.0.0"
 toposort = ">1.0,<2.0"