From 349e03a0060c09071dd6777a7a770eb30e5c8bca Mon Sep 17 00:00:00 2001 From: Chenyu Li Date: Fri, 3 May 2024 12:46:34 -0700 Subject: [PATCH] nits --- core/dbt/deps/resolver.py | 2 +- tests/functional/configs/test_custom_node_colors_configs.py | 3 ++- 2 files changed, 3 insertions(+), 2 deletions(-) diff --git a/core/dbt/deps/resolver.py b/core/dbt/deps/resolver.py index f613d649809..4c575b9345e 100644 --- a/core/dbt/deps/resolver.py +++ b/core/dbt/deps/resolver.py @@ -5,9 +5,9 @@ from dbt.config.renderer import PackageRenderer from dbt.contracts.project import ( GitPackage, - PrivatePackage, LocalPackage, PackageSpec, + PrivatePackage, RegistryPackage, TarballPackage, ) diff --git a/tests/functional/configs/test_custom_node_colors_configs.py b/tests/functional/configs/test_custom_node_colors_configs.py index 1ca28d67795..7772e3d44ca 100644 --- a/tests/functional/configs/test_custom_node_colors_configs.py +++ b/tests/functional/configs/test_custom_node_colors_configs.py @@ -1,5 +1,6 @@ import pytest +from dbt.exceptions import ConfigUpdateError from dbt.tests.util import get_manifest, run_dbt from dbt_common.dataclass_schema import ValidationError @@ -304,7 +305,7 @@ def test__invalid_color_config_block( self, project, ): - with pytest.raises(ValidationError): + with pytest.raises((ValidationError, ConfigUpdateError)): run_dbt(["compile"])