industrial-model 0.1.7__tar.gz → 0.1.9__tar.gz

{industrial_model-0.1.7 → industrial_model-0.1.9}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: industrial-model
-Version: 0.1.7
+Version: 0.1.9
 Summary: Industrial Model ORM
 Author-email: Lucas Alves <lucasrosaalves@gmail.com>
 Classifier: Programming Language :: Python

{industrial_model-0.1.7 → industrial_model-0.1.9}/industrial_model/cognite_adapters/query_result_mapper.py

@@ -70,13 +70,14 @@ class QueryResultMapper:
 
             return entry.get("space", ""), entry.get("externalId", "")
 
-        nodes = {
-            (node.space, node.external_id): node
-            for node in query_result[key]
-            if isinstance(node, Node)
-        }
+        visited: set[tuple[str, str]] = set()
         result: defaultdict[tuple[str, str], list[Node]] = defaultdict(list)
-        for node in nodes.values():
+        for node in query_result[key]:
+            identify = (node.space, node.external_id)
+            if not isinstance(node, Node) or identify in visited:
+                continue
+
+            visited.add(identify)
             properties = node.properties.get(view_id, {})
             node_id = get_node_id(node)
             for mapping_key, (values, is_list) in mappings.items():
@@ -85,7 +86,7 @@ class QueryResultMapper:
                 element_key: tuple[str, str] = (
                     (element.get("space", ""), element.get("externalId", ""))
                     if isinstance(element, dict)
-                    else node_id
+                    else (node.space, node.external_id)
                 )
 
                 entries = values.get(element_key)
@@ -178,14 +179,14 @@ class QueryResultMapper:
         if not nodes:
             return None
 
+        visited: set[tuple[str, str]] = set()
         result: defaultdict[tuple[str, str], list[Node]] = defaultdict(list)
+        for edge in query_result[edge_key]:
+            identify = (edge.space, edge.external_id)
+            if not isinstance(edge, Edge) or identify in visited:
+                continue
 
-        edges = {
-            (edge.space, edge.external_id): edge
-            for edge in query_result[edge_key]
-            if isinstance(edge, Edge)
-        }
-        for edge in edges.values():
+            visited.add(identify)
             entry_key, node_key = (
                 (
                     edge.end_node.as_tuple(),

{industrial_model-0.1.7 → industrial_model-0.1.9}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "industrial-model"
-version = "0.1.7"
+version = "0.1.9"
 description = "Industrial Model ORM"
 readme = "README.md"
 requires-python = ">=3.11"

{industrial_model-0.1.7 → industrial_model-0.1.9}/uv.lock

@@ -215,7 +215,7 @@ wheels = [
 
 [[package]]
 name = "industrial-model"
-version = "0.1.7"
+version = "0.1.9"
 source = { editable = "." }
 dependencies = [
     { name = "anyio" },