[PipelineAdapters] Ensure warning is suppressed (#5576)

mlrun · May 16, 2024 · 9227bc2 · 9227bc2
1 parent b3bc6f6
commit 9227bc2
Show file tree

Hide file tree

Showing 6 changed files with 14 additions and 1 deletion.
diff --git a/pipeline-adapters/mlrun-pipelines-kfp-v1-8/setup.py b/pipeline-adapters/mlrun-pipelines-kfp-v1-8/setup.py
@@ -21,7 +21,7 @@
 
 setup(
     name="mlrun-pipelines-kfp-v1-8-experiment",
-    version="0.2.1",
+    version="0.2.2",
     description="MLRun Pipelines package for providing KFP 1.8 compatibility",
     author="Yaron Haviv",
     author_email="yaronh@iguazio.com",

diff --git a/pipeline-adapters/mlrun-pipelines-kfp-v1-8/src/mlrun_pipelines/mixins.py b/pipeline-adapters/mlrun-pipelines-kfp-v1-8/src/mlrun_pipelines/mixins.py
@@ -19,6 +19,9 @@
 
 import mlrun
 
+# Disable the warning about reusing components
+kfp.dsl.ContainerOp._DISABLE_REUSABLE_COMPONENT_WARNING = True
+
 
 class KfpAdapterMixin:
     def apply(self, modify):

diff --git a/pipeline-adapters/mlrun-pipelines-kfp-v1-8/src/mlrun_pipelines/mounts.py b/pipeline-adapters/mlrun-pipelines-kfp-v1-8/src/mlrun_pipelines/mounts.py
@@ -24,6 +24,9 @@
 from mlrun.platforms.iguazio import v3io_to_vol
 from mlrun.utils import logger
 
+# Disable the warning about reusing components
+kfp.dsl.ContainerOp._DISABLE_REUSABLE_COMPONENT_WARNING = True
+
 
 def v3io_cred(api="", user="", access_key=""):
     """

diff --git a/pipeline-adapters/mlrun-pipelines-kfp-v1-8/src/mlrun_pipelines/ops.py b/pipeline-adapters/mlrun-pipelines-kfp-v1-8/src/mlrun_pipelines/ops.py
@@ -31,6 +31,7 @@
 from mlrun.config import config
 from mlrun.utils import get_in
 
+# Disable the warning about reusing components
 dsl.ContainerOp._DISABLE_REUSABLE_COMPONENT_WARNING = True
 
 

diff --git a/pipeline-adapters/mlrun-pipelines-kfp-v1-8/src/mlrun_pipelines/patcher.py b/pipeline-adapters/mlrun-pipelines-kfp-v1-8/src/mlrun_pipelines/patcher.py
@@ -22,6 +22,9 @@
 from mlrun.errors import err_to_str
 from mlrun.utils import logger
 
+# Disable the warning about reusing components
+kfp.dsl.ContainerOp._DISABLE_REUSABLE_COMPONENT_WARNING = True
+
 
 # When we run pipelines, the kfp.compile.Compile.compile() method takes the decorated function with @dsl.pipeline and
 # converts it to a k8s object. As part of the flow in the Compile.compile() method,

diff --git a/pipeline-adapters/mlrun-pipelines-kfp-v1-8/src/mlrun_pipelines/utils.py b/pipeline-adapters/mlrun-pipelines-kfp-v1-8/src/mlrun_pipelines/utils.py
@@ -18,6 +18,9 @@
 from kubernetes import client
 from mlrun_pipelines.helpers import new_pipe_metadata
 
+# Disable the warning about reusing components
+kfp.dsl.ContainerOp._DISABLE_REUSABLE_COMPONENT_WARNING = True
+
 
 def apply_kfp(modify, cop, runtime):
     modify(cop)