update type for mashumaro, update related test

dbt-labs · Aug 31, 2023 · 8b643bd · 8b643bd
1 parent a7db607
commit 8b643bd
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 6 deletions.
diff --git a/core/dbt/adapters/base/relation.py b/core/dbt/adapters/base/relation.py
@@ -11,7 +11,6 @@
     Policy,
     Path,
 )
-from dbt.dataclass_schema import StrEnum
 from dbt.exceptions import (
     ApproximateMatchError,
     DbtInternalError,
@@ -37,7 +36,7 @@ class BaseRelation(FakeAPIObject, Hashable):
     quote_policy: Policy = field(default_factory=lambda: Policy())
     dbt_created: bool = False
     # register relation types that can be renamed for the purpose of replacing relations using stages and backups
-    renameable_relations: List[StrEnum] = field(
+    renameable_relations: List[str] = field(
         default_factory=lambda: [RelationType.Table, RelationType.View]
     )
 

diff --git a/tests/unit/test_relation.py b/tests/unit/test_relation.py
@@ -1,5 +1,3 @@
-from dataclasses import replace
-
 import pytest
 
 from dbt.adapters.base import BaseRelation
@@ -10,10 +8,9 @@
     "relation_type,result",
     [
         (RelationType.View, True),
-        (RelationType.Table, False),
+        (RelationType.External, False),
     ],
 )
 def test_can_be_renamed(relation_type, result):
     my_relation = BaseRelation.create(type=relation_type)
-    my_relation = replace(my_relation, renameable_relations=[RelationType.View])
     assert my_relation.can_be_renamed is result