get rid of pydantic warnings

databricks · Nov 13, 2024 · 46db57d · 46db57d
1 parent e33c704
commit 46db57d
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 1 deletion.
diff --git a/requirements.txt b/requirements.txt
@@ -6,3 +6,4 @@ dbt-adapters>=1.7.0, <2.0
 databricks-sdk==0.17.0
 keyring>=23.13.0
 protobuf<5.0.0
+pydantic>=1.10.0, <2
diff --git a/setup.py b/setup.py
@@ -63,6 +63,7 @@ def _get_plugin_version() -> str:
         "keyring>=23.13.0",
         "pandas<2.2.0",
         "protobuf<5.0.0",
+        "pydantic>=1.10.0, <2",
     ],
     zip_safe=False,
     classifiers=[

diff --git a/tests/unit/python/test_python_config.py b/tests/unit/python/test_python_config.py
@@ -101,7 +101,7 @@ def test_parsed_model__run_name(self):
         assert model.run_name.startswith("hive_metastore-default-test-")
 
     def test_parsed_model__invalid_config(self):
-        parsed_model = {"alias": "test", "config": []}
+        parsed_model = {"alias": "test", "config": 1}
         with pytest.raises(ValidationError):
             ParsedPythonModel(**parsed_model)