diff --git a/pyproject.toml b/pyproject.toml index 5cff256b28..1c32bab70e 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -47,11 +47,15 @@ classifiers = [ dependencies = ["databricks-sdk>=0.58.0,<0.59.0", "databricks-labs-lsql>=0.16.0,<0.17.0", - "databricks-labs-blueprint>=0.11.0,<0.12.0", + #"databricks-labs-blueprint>=0.11.0,<0.12.0", + "databricks-labs-blueprint @ git+https://github.com/databrickslabs/blueprint@patch/bool_type", "PyYAML>=6.0.0,<6.1.0", "sqlglot>=26.7.0,<27.1.0", "astroid>=3.3.0,<3.4.0"] +[tool.hatch.metadata] +allow-direct-references = true + [project.optional-dependencies] pylsp = [ "python-lsp-server>=1.13.0,<1.14.0" diff --git a/tests/unit/hive_metastore/test_mapping.py b/tests/unit/hive_metastore/test_mapping.py index 01944256f6..a495a9d093 100644 --- a/tests/unit/hive_metastore/test_mapping.py +++ b/tests/unit/hive_metastore/test_mapping.py @@ -137,6 +137,7 @@ def test_save_mapping(): [ { 'catalog_name': 'foo_bar', + 'compare_rows': False, 'dst_schema': 'foo', 'dst_table': 'bar', 'src_schema': 'foo', diff --git a/tests/unit/install/test_install.py b/tests/unit/install/test_install.py index 9c9c70ee11..7ef52e1a79 100644 --- a/tests/unit/install/test_install.py +++ b/tests/unit/install/test_install.py @@ -379,6 +379,12 @@ def test_configure_sets_expected_workspace_configuration_values( 'query_statement_disposition': 'INLINE', "recon_tolerance_percent": 5, "managed_table_external_storage": "CLONE", + "is_terraform_used": False, + "trigger_job": False, + "upload_dependencies": False, + "enable_hms_federation": False, + "use_legacy_permission_migration": False, + "skip_tacl_migration": False, } prompts = MockPrompts( { @@ -419,6 +425,12 @@ def test_configure_with_default_owner_group( "num_days_submit_runs_history": 30, "recon_tolerance_percent": 5, "managed_table_external_storage": "CLONE", + "is_terraform_used": False, + "trigger_job": False, + "upload_dependencies": False, + "enable_hms_federation": False, + "use_legacy_permission_migration": False, + "skip_tacl_migration": False, } prompts = MockPrompts( { @@ -509,6 +521,12 @@ def test_create_cluster_policy(ws, mock_installation) -> None: 'query_statement_disposition': 'INLINE', 'recon_tolerance_percent': 5, 'managed_table_external_storage': "CLONE", + 'is_terraform_used': False, + 'trigger_job': False, + 'upload_dependencies': False, + 'enable_hms_federation': False, + 'use_legacy_permission_migration': False, + 'skip_tacl_migration': False, }, ) @@ -1753,6 +1771,12 @@ def test_save_config_ext_hms(ws, mock_installation) -> None: 'query_statement_disposition': 'INLINE', 'recon_tolerance_percent': 5, 'managed_table_external_storage': "CLONE", + 'is_terraform_used': False, + 'trigger_job': False, + 'upload_dependencies': False, + 'enable_hms_federation': False, + 'use_legacy_permission_migration': False, + 'skip_tacl_migration': False, }, )