Fix tests

dagster-io · Nov 27, 2024 · 67d0fcd · 67d0fcd
1 parent 9b0b627
commit 67d0fcd
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 6 deletions.
diff --git a/...odules/libraries/dagster-fivetran/dagster_fivetran_tests/experimental/test_asset_specs.py b/...odules/libraries/dagster-fivetran/dagster_fivetran_tests/experimental/test_asset_specs.py
@@ -1,6 +1,6 @@
 import responses
 from dagster._config.field_utils import EnvVar
-from dagster._core.definitions.asset_spec import AssetSpec, replace_attributes
+from dagster._core.definitions.asset_spec import AssetSpec
 from dagster._core.test_utils import environ
 from dagster_fivetran import (
     DagsterFivetranTranslator,
@@ -123,8 +123,7 @@ def test_cached_load_spec_with_asset_factory(
 class MyCustomTranslator(DagsterFivetranTranslator):
     def get_asset_spec(self, data: FivetranConnectorTableProps) -> AssetSpec:
         default_spec = super().get_asset_spec(data)
-        return replace_attributes(
-            default_spec,
+        return default_spec.replace_attributes(
             key=default_spec.key.with_prefix("prefix"),
             metadata={**default_spec.metadata, "custom": "metadata"},
         )

diff --git a/...modules/libraries/dagster-fivetran/dagster_fivetran_tests/experimental/test_translator.py b/...modules/libraries/dagster-fivetran/dagster_fivetran_tests/experimental/test_translator.py
@@ -2,7 +2,7 @@
 
 import responses
 from dagster._config.field_utils import EnvVar
-from dagster._core.definitions.asset_spec import AssetSpec, replace_attributes
+from dagster._core.definitions.asset_spec import AssetSpec
 from dagster._core.test_utils import environ
 from dagster_fivetran import (
     DagsterFivetranTranslator,
@@ -36,8 +36,7 @@ def test_fivetran_workspace_data_to_fivetran_connector_table_props_data(
 class MyCustomTranslator(DagsterFivetranTranslator):
     def get_asset_spec(self, props: FivetranConnectorTableProps) -> AssetSpec:
         default_spec = super().get_asset_spec(props)
-        return replace_attributes(
-            default_spec,
+        return default_spec.replace_attributes(
             key=default_spec.key.with_prefix("prefix"),
             metadata={**default_spec.metadata, "custom": "metadata"},
         )