node.resource_type

AltimateAI · May 30, 2024 · 4145be8 · 4145be8
1 parent b157547
commit 4145be8
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/src/datapilot/core/platforms/dbt/wrappers/manifest/v12/wrapper.py b/src/datapilot/core/platforms/dbt/wrappers/manifest/v12/wrapper.py
@@ -60,7 +60,7 @@ def _get_node(self, node: ManifestNode) -> AltimateManifestNode:
             language,
             contract,
         ) = ([], [], None, None, None, None, None, "", "", None)
-        if node.resource_type.value != SEED:
+        if node.resource_type != SEED:
             sources = node.sources
             metrics = node.metrics
             depends_on_nodes = node.depends_on.nodes if node.depends_on else None
@@ -75,7 +75,7 @@ def _get_node(self, node: ManifestNode) -> AltimateManifestNode:
             database=node.database,
             schema_name=node.schema_,
             name=node.name,
-            resource_type=AltimateResourceType(node.resource_type.value),
+            resource_type=AltimateResourceType(node.resource_type),
             package_name=node.package_name,
             path=node.path,
             description=node.description,
@@ -337,7 +337,7 @@ def get_nodes(
         nodes = {}
         for node in self.manifest.nodes.values():
             if (
-                node.resource_type.value
+                node.resource_type
                 in [
                     AltimateResourceType.seed.value,
                     AltimateResourceType.test.value,
@@ -381,7 +381,7 @@ def get_tests(self, type=None) -> Dict[str, AltimateManifestTestNode]:
 
         for node in self.manifest.nodes.values():
             # Check if the node is a test and of the correct type
-            if node.resource_type.value == AltimateResourceType.test.value:
+            if node.resource_type == AltimateResourceType.test.value:
                 if any(isinstance(node, t) for t in types):
                     tests[node.unique_id] = self._get_tests(node)
         return tests