Support Kedro 1.x while preserving 0.19 compatibility

- widen the Kedro dependency from `>0.18.4, <=0.19.9` to `>0.18.4, <2.0`
  so projects can upgrade to Kedro 1.x to pick up security fixes
- keep backward compatibility for Kedro 0.19 via an AbstractConfigLoader shim
- remove dead Kedro 0.16 compatibility code and the unused semver dependency
- replace the Pydantic v1-only deep_update import in tests with a stdlib helper
- update docs and changelog for the revised Kedro compatibility story
- make MLflow-related tests self-contained so the default test suite passes
  without optional mlflow/kedro-mlflow packages installed

Author: rdziadosz

CHANGELOG.md

@@ -2,6 +2,24 @@
 
 ## [Unreleased]
 
+### Changed
+
+- Widened Kedro dependency to `>0.18.4, <2.0` to support both Kedro 0.19.x and 1.x
+- Updated minimum Python version to 3.9 and upper bound to <3.13
+- Added compatibility shim for `AbstractConfigLoader` removal in Kedro 1.x
+- Updated documentation: dynamic configuration examples now use `OmegaConfigLoader` with `oc.env` resolver instead of deprecated `TemplatedConfigLoader`
+
+### Removed
+
+- Removed `ContextHelper16` class (Kedro 0.16 compatibility — dead code since minimum is 0.18.4)
+- Removed `semver` dependency (only used for obsolete version check)
+
+### Fixed
+
+- Replaced Pydantic v1-only `pydantic.utils.deep_update` import in tests with stdlib implementation
+- Fixed internal module path in `test_kfpclient.py` for `KedroContext` mock patch
+- Fixed mlflow-related tests to not require `mlflow`/`kedro_mlflow` packages installed (using `sys.modules` stubbing)
+
 ## [0.10.0] - 2026-04-27
 
 - Add support for `kedro 0.19.9`

docs/conf.py

@@ -28,7 +28,7 @@
 author = "GetInData"
 
 myst_substitutions = {
-    "tested_kedro": "0.17.7",
+    "tested_kedro": "1.0.0",
     "release": release,
 }
 

docs/source/02_installation/02_configuration.md

@@ -176,12 +176,14 @@ run_config:
 
 ## Dynamic configuration support
 
-`kedro-kubeflow` contains hook that enables [TemplatedConfigLoader](https://kedro.readthedocs.io/en/stable/kedro.config.TemplatedConfigLoader.html).
-It allows passing environment variables to configuration files. It reads all environment variables following `KEDRO_CONFIG_<NAME>` pattern, which you 
-can later inject in configuration file using `${name}` syntax. 
+`kedro-kubeflow` supports dynamic configuration via Kedro's `OmegaConfigLoader` with the `oc.env` resolver.
+It allows passing environment variables to configuration files. It reads all environment variables following `KEDRO_CONFIG_<NAME>` pattern, which you
+can later inject in configuration file using `${oc.env:KEDRO_CONFIG_NAME}` syntax.
 
-There are two special variables `KEDRO_CONFIG_COMMIT_ID`, `KEDRO_CONFIG_BRANCH_NAME` with support specifying default when variable is not set, 
-e.g. `${commit_id|dirty}`   
+There are two special variables `KEDRO_CONFIG_COMMIT_ID`, `KEDRO_CONFIG_BRANCH_NAME` with support specifying default when variable is not set,
+e.g. `${oc.env:KEDRO_CONFIG_COMMIT_ID, dirty}`
+
+> **Note:** The legacy `EnvTemplatedConfigLoader` is deprecated and will be removed in a future release. Please migrate to `OmegaConfigLoader` with `oc.env` resolver.   
 
 ## Extra volumes
 You can mount additional volumes (such as `emptyDir`) to specific nodes by using `extra_volumes` config node.

kedro_kubeflow/context_helper.py

@@ -2,16 +2,21 @@
 from functools import cached_property, lru_cache
 from typing import Any, Dict
 
-from kedro import __version__ as kedro_version
 from kedro.config import (
-    AbstractConfigLoader,
     MissingConfigException,
     OmegaConfigLoader,
 )
+
+try:
+    from kedro.config import AbstractConfigLoader
+except ImportError:
+    # AbstractConfigLoader was removed in Kedro 1.0;
+    # OmegaConfigLoader is the only config loader from that version on.
+    AbstractConfigLoader = OmegaConfigLoader
+
 from kedro.framework.session import KedroSession
 from omegaconf import DictConfig, OmegaConf
 from omegaconf.resolvers import oc
-from semver import VersionInfo
 
 from .config import PluginConfig
 
@@ -146,20 +151,4 @@ def kfp_client(self):
 
     @staticmethod
     def init(metadata, env):
-        version = VersionInfo.parse(kedro_version)
-        if version.match(">=0.17.0"):
-            return ContextHelper(metadata, env)
-        else:
-            return ContextHelper16(metadata, env)
-
-
-class ContextHelper16(ContextHelper):
-    """KedroKubeflowConfig vairant for compatibility with Kedro 1.6"""
-
-    @property
-    def project_name(self):
-        return self.context.project_path.name
-
-    @property
-    def context(self):
-        return self.session.load_context()
+        return ContextHelper(metadata, env)

poetry.lock

@@ -34,16 +34,15 @@ line-length = 121
 line-length = 121
 
 [tool.isort]
-known_third_party = ["click", "google", "kedro", "kfp", "kubernetes", "tabulate", "pydantic", "semver", "setuptools"]
+known_third_party = ["click", "google", "kedro", "kfp", "kubernetes", "tabulate", "pydantic", "setuptools"]
 
 [tool.poetry.dependencies]
-python = ">=3.8,<3.12"
-kedro = ">0.18.4, <=0.19.9"
+python = ">=3.9,<3.13"
+kedro = ">0.18.4, <2.0"
 # kedro-viz = { version = "<8", optional = true} # because of pydantic conflict
 click = ">=8.0.4"
 kfp = ">=1.8.12,<2.0" # blocking pydantic upgrade
 tabulate = ">=0.8.7"
-semver = "~=2.10"
 fsspec = ">=2021.4"
 pyyaml = ">=6.0,<7.0"
 google-auth = { version = "<3", optional = true}

pyproject.toml

@@ -1,4 +1,14 @@
-from pydantic.utils import deep_update
+import copy
+
+
+def deep_update(mapping, updating_mapping):
+    updated_mapping = copy.deepcopy(mapping)
+    for k, v in updating_mapping.items():
+        if k in updated_mapping and isinstance(updated_mapping[k], dict) and isinstance(v, dict):
+            updated_mapping[k] = deep_update(updated_mapping[k], v)
+        else:
+            updated_mapping[k] = v
+    return updated_mapping
 
 
 class MinimalConfigMixin:

tests/common.py

@@ -270,19 +270,17 @@ def test_init_with_github_actions(self, cwd):
                 assert "kedro kubeflow upload-pipeline" in content
                 assert "kedro kubeflow schedule" in content
 
-    @patch("mlflow.start_run")
-    @patch("mlflow.set_tag")
-    @patch("mlflow.get_experiment_by_name")
-    def test_mlflow_start(self, get_experiment_by_name_mock, set_tag_mock, start_run_mock):
+    def test_mlflow_start(self):
         context_helper = MagicMock(ContextHelper)
         config = dict(context_helper=context_helper)
         runner = CliRunner()
-        get_experiment_by_name_mock.return_value = type("obj", (object,), {"experiment_id": 47})
-        start_run_mock.return_value = namedtuple("InfoObject", "info")(
+        mlflow_mock = MagicMock()
+        mlflow_mock.get_experiment_by_name.return_value = type("obj", (object,), {"experiment_id": 47})
+        mlflow_mock.start_run.return_value = namedtuple("InfoObject", "info")(
             namedtuple("RunIdObject", "run_id")("MLFLOW_RUN_ID")
         )
 
-        with TemporaryDirectory() as temp_dir:
+        with patch.dict("sys.modules", {"mlflow": mlflow_mock}), TemporaryDirectory() as temp_dir:
             run_id_file_path = f"{temp_dir}/run_id"
             result = runner.invoke(
                 mlflow_start,
@@ -295,7 +293,7 @@ def test_mlflow_start(self, get_experiment_by_name_mock, set_tag_mock, start_run
             with open(run_id_file_path) as f:
                 assert f.read() == "MLFLOW_RUN_ID"
 
-        set_tag_mock.assert_called_with("kubeflow_run_id", "KUBEFLOW_RUN_ID")
+        mlflow_mock.set_tag.assert_called_with("kubeflow_run_id", "KUBEFLOW_RUN_ID")
 
     @patch("kubernetes.client")
     @patch("kubernetes.config")

tests/test_cli.py

@@ -11,7 +11,6 @@
 from kedro_kubeflow.config import PluginConfig
 from kedro_kubeflow.context_helper import (
     ContextHelper,
-    ContextHelper16,
     EnvTemplatedConfigLoader,
 )
 
@@ -20,11 +19,6 @@
 
 
 class TestContextHelper(unittest.TestCase, MinimalConfigMixin):
-    def test_init_different_kedro_versions(self):
-
-        with patch("kedro_kubeflow.context_helper.kedro_version", "0.16.0"):
-            ch = ContextHelper.init(None, None)
-            assert isinstance(ch, ContextHelper16)
 
     def test_project_name(self):
         metadata = Mock()

tests/test_context_helper.py

@@ -1,8 +1,6 @@
 import os
 import unittest
-from unittest.mock import patch
-
-import mlflow
+from unittest.mock import MagicMock, patch
 
 from kedro_kubeflow.auth import AuthHandler
 from kedro_kubeflow.hooks import MlflowIapAuthHook, MlflowTagsHook  # NOQA
@@ -19,43 +17,39 @@ def test_should_inject_token_when_env_is_set(self, obtain_id_token):
         obtain_id_token.assert_called_with()
 
 
-@patch.object(mlflow, "set_tag")
 class TestMlflowTagsHook(unittest.TestCase):
-    def test_should_set_mlflow_tags(self, mlflow_set_tag):
-        with environment({"KUBEFLOW_RUN_ID": "KFP_123"}):
+    def test_should_set_mlflow_tags(self):
+        mlflow_mock = MagicMock()
+        with patch.dict("sys.modules", {"mlflow": mlflow_mock, "kedro_mlflow": MagicMock()}), environment(
+            {"KUBEFLOW_RUN_ID": "KFP_123"}
+        ):
             MlflowTagsHook().before_node_run()
 
-        mlflow_set_tag.assert_called_with("kubeflow_run_id", "KFP_123")
+        mlflow_mock.set_tag.assert_called_with("kubeflow_run_id", "KFP_123")
 
-    def test_should_not_set_mlflow_tags_when_kubeflow_run_id_env_is_not_set(self, mlflow_set_tag):
-        with environment({}, delete_keys=["KUBEFLOW_RUN_ID"]):
+    def test_should_not_set_mlflow_tags_when_kubeflow_run_id_env_is_not_set(self):
+        mlflow_mock = MagicMock()
+        with patch.dict("sys.modules", {"mlflow": mlflow_mock, "kedro_mlflow": MagicMock()}), environment(
+            {}, delete_keys=["KUBEFLOW_RUN_ID"]
+        ):
             MlflowTagsHook().before_node_run()
 
-        mlflow_set_tag.assert_not_called()
+        mlflow_mock.set_tag.assert_not_called()
 
-    def test_should_not_set_mlflow_tags_when_kubeflow_run_id_env_is_empty(self, mlflow_set_tag):
-        with environment({"KUBEFLOW_RUN_ID": ""}):
+    def test_should_not_set_mlflow_tags_when_kubeflow_run_id_env_is_empty(self):
+        mlflow_mock = MagicMock()
+        with patch.dict("sys.modules", {"mlflow": mlflow_mock, "kedro_mlflow": MagicMock()}), environment(
+            {"KUBEFLOW_RUN_ID": ""}
+        ):
             MlflowTagsHook().before_node_run()
 
-        mlflow_set_tag.assert_not_called()
-
-    def test_should_not_set_mlflow_tags_when_mlflow_is_not_enabled(self, mlflow_set_tag):
-        # given
-        real_import = __builtins__["__import__"]
-
-        def mlflow_import_disabled(name, *args, **kw):
-            if name == "mlflow":
-                raise ImportError
-            return real_import(name, *args, **kw)
+        mlflow_mock.set_tag.assert_not_called()
 
-        __builtins__["__import__"] = mlflow_import_disabled
-
-        # when
+    @patch("kedro_kubeflow.hooks.is_mlflow_enabled", return_value=False)
+    def test_should_not_set_mlflow_tags_when_mlflow_is_not_enabled(self, _):
+        mlflow_mock = MagicMock()
         with environment({"KUBEFLOW_RUN_ID": "KFP_123"}):
-            MlflowTagsHook().before_node_run()
-
-        # then
-        mlflow_set_tag.assert_not_called()
+            with patch.dict("sys.modules", {"mlflow": mlflow_mock}):
+                MlflowTagsHook().before_node_run()
 
-        # cleanup
-        __builtins__["__import__"] = real_import
+        mlflow_mock.set_tag.assert_not_called()