fix(sdk): fixes dsl.ContainerOp deprecation warning not shown (#4658)

* change dsl.ContainerOp warning to FutureWarning * fix tests
kubeflow · Oct 24, 2020 · c52a81c · c52a81c
1 parent d7793af
commit c52a81c
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/sdk/python/kfp/dsl/_container_op.py b/sdk/python/kfp/dsl/_container_op.py
@@ -1025,7 +1025,7 @@ def __init__(
                 " The components can be created manually (or, in case of python, using kfp.components.create_component_from_func or func_to_container_op)"
                 " and then loaded using kfp.components.load_component_from_file, load_component_from_uri or load_component_from_text: "
                 "https://kubeflow-pipelines.readthedocs.io/en/latest/source/kfp.components.html#kfp.components.load_component_from_file",
-                category=DeprecationWarning,
+                category=FutureWarning,
             )
 
         self.attrs_with_pipelineparams = BaseOp.attrs_with_pipelineparams + ['_container', 'artifact_arguments', '_parameter_arguments'] #Copying the BaseOp class variable!

diff --git a/sdk/python/tests/dsl/component_bridge_tests.py b/sdk/python/tests/dsl/component_bridge_tests.py
@@ -241,7 +241,7 @@ def test_reusable_component_warnings(self):
             deprecation_messages = list(str(message) for message in warning_messages if message.category == DeprecationWarning)
             self.assertListEqual(deprecation_messages, [])
 
-        with self.assertWarnsRegex(DeprecationWarning, expected_regex='reusable'):
+        with self.assertWarnsRegex(FutureWarning, expected_regex='reusable'):
             kfp.dsl.ContainerOp(name='name', image='image')
 
     def test_prevent_passing_container_op_as_argument(self):