diff --git a/clients/python/src/kaskada/api/release.py b/clients/python/src/kaskada/api/release.py index ee6f3bcb9..f61d3660f 100644 --- a/clients/python/src/kaskada/api/release.py +++ b/clients/python/src/kaskada/api/release.py @@ -14,6 +14,7 @@ logging.basicConfig(stream=sys.stdout, level=logging.INFO) logger = logging.getLogger(__name__) + class LocalRelease(object): """Configuration details for a local release. Mostly used for binary paths.""" diff --git a/clients/python/src/kaskada/api/session.py b/clients/python/src/kaskada/api/session.py index 6c47e4ea9..8364fce81 100644 --- a/clients/python/src/kaskada/api/session.py +++ b/clients/python/src/kaskada/api/session.py @@ -28,6 +28,7 @@ KASKADA_DISABLE_DOWNLOAD_ENV = "KASKADA_DISABLE_DOWNLOAD" + class Session: def __init__( self, diff --git a/clients/python/tests/api/test_session.py b/clients/python/tests/api/test_session.py index a8e64ca02..86a43c2c9 100644 --- a/clients/python/tests/api/test_session.py +++ b/clients/python/tests/api/test_session.py @@ -82,14 +82,16 @@ def test_local_builder_set_engine_version(): version = "engine@v0.0.1-beta.1" builder = kaskada.api.session.LocalBuilder().engine_version(version) assert builder._engine_version == version - + + def test_local_builder_set_engine_version_throws(): invalid_versions = ["0.0.1", "engine@0.0.1", "engine@v23", "manager@v1.1.1"] - + for version in invalid_versions: with pytest.raises(ValueError): builder = kaskada.api.session.LocalBuilder().engine_version(version) + def test_local_builder_set_path_sets_path(): path = "my_path" builder = kaskada.api.session.LocalBuilder().path(path)