cognite-neat 0.116.0__tar.gz → 0.117.0__tar.gz

{cognite_neat-0.116.0 → cognite_neat-0.117.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: cognite-neat
-Version: 0.116.0
+Version: 0.117.0
 Summary: Knowledge graph transformation
 License: Apache-2.0
 Author: Nikola Vasiljevic
@@ -18,7 +18,7 @@ Provides-Extra: google
 Provides-Extra: oxi
 Requires-Dist: PyYAML
 Requires-Dist: backports.strenum (>=1.2,<2.0) ; python_version < "3.11"
-Requires-Dist: cognite-sdk (>=7.71.2,<8.0.0)
+Requires-Dist: cognite-sdk (>=7.73.4,<8.0.0)
 Requires-Dist: elementpath (>=4.0.0,<5.0.0)
 Requires-Dist: exceptiongroup (>=1.1.3,<2.0.0) ; python_version < "3.11"
 Requires-Dist: google-api-python-client ; extra == "google"

{cognite_neat-0.116.0 → cognite_neat-0.117.0}/cognite/neat/_graph/extractors/_dms.py

@@ -215,23 +215,23 @@ class _ViewInstanceIterator(Iterable[Instance]):
     def count(self) -> int:
         node_count = edge_count = 0
         if self.view.used_for in ("node", "all"):
-            node_count = int(
-                self.client.data_modeling.instances.aggregate(
-                    view=self.view.as_id(),
-                    aggregates=dm.aggregations.Count("externalId"),
-                    instance_type="node",
-                    space=self.instance_space,
-                ).value
-            )
+            node_result = self.client.data_modeling.instances.aggregate(
+                view=self.view.as_id(),
+                aggregates=dm.aggregations.Count("externalId"),
+                instance_type="node",
+                space=self.instance_space,
+            ).value
+            if node_result:
+                node_count = int(node_result)
         if self.view.used_for in ("edge", "all"):
-            edge_count = int(
-                self.client.data_modeling.instances.aggregate(
-                    view=self.view.as_id(),
-                    aggregates=dm.aggregations.Count("externalId"),
-                    instance_type="edge",
-                    space=self.instance_space,
-                ).value
-            )
+            edge_result = self.client.data_modeling.instances.aggregate(
+                view=self.view.as_id(),
+                aggregates=dm.aggregations.Count("externalId"),
+                instance_type="edge",
+                space=self.instance_space,
+            ).value
+            if edge_result:
+                edge_count = int(edge_result)
         return node_count + edge_count
 
     def __iter__(self) -> Iterator[Instance]:

{cognite_neat-0.116.0 → cognite_neat-0.117.0}/cognite/neat/_session/_base.py

@@ -23,7 +23,6 @@ from cognite.neat._store._rules_store import RulesEntity
 from cognite.neat._utils.auxiliary import local_import
 
 from ._collector import _COLLECTOR, Collector
-from ._create import CreateAPI
 from ._drop import DropAPI
 from ._explore import ExploreAPI
 from ._fix import FixAPI
@@ -35,6 +34,7 @@ from ._set import SetAPI
 from ._show import ShowAPI
 from ._state import SessionState
 from ._subset import SubsetAPI
+from ._template import TemplateAPI
 from ._to import ToAPI
 from .engine import load_neat_engine
 from .exceptions import session_class_wrapper
@@ -104,7 +104,7 @@ class NeatSession:
         self.mapping = MappingAPI(self._state)
         self.drop = DropAPI(self._state)
         self.subset = SubsetAPI(self._state)
-        self.create = CreateAPI(self._state)
+        self.template = TemplateAPI(self._state)
         self._explore = ExploreAPI(self._state)
         self.opt = OptAPI()
         self.opt._display()

cognite_neat-0.116.0/cognite/neat/_session/_create.py → cognite_neat-0.117.0/cognite/neat/_session/_template.py

@@ -17,9 +17,9 @@ from .exceptions import NeatSessionError, session_class_wrapper
 
 
 @session_class_wrapper
-class CreateAPI:
+class TemplateAPI:
     """
-    Create new data model based on the given data.
+    Create a template for a new data model.
     """
 
     def __init__(self, state: SessionState):
@@ -31,7 +31,10 @@ class CreateAPI:
         org_name: str = "CopyOf",
         dummy_property: str = "GUID",
     ) -> IssueList:
-        """Uses the current data model as a basis to create enterprise data model
+        """Creates a template for an enterprise model based on the current data model in the session.
+        An enterprise data model is a model that is used for read and write of instances. In addition,
+        it is governed by the organization.
+        The basis for an enterprise data model should be a Cognite Data Model.
 
         Args:
             data_model_id: The enterprise data model id that is being created
@@ -76,7 +79,9 @@ class CreateAPI:
         direct_property: str = "enterprise",
         view_prefix: str = "Enterprise",
     ) -> IssueList:
-        """Uses the current data model as a basis to create solution data model
+        """Creates a template for a solution model based on the current data model in the session.
+        A solution data model is for read and write of instances.
+        The basis for a solution data model should be an enterprise data model.
 
         Args:
             data_model_id: The solution data model id that is being created.
@@ -121,7 +126,9 @@ class CreateAPI:
         data_model_id: DataModelIdentifier,
         include: Literal["same-space", "all"] = "same-space",
     ) -> IssueList:
-        """Uses the current data model as a basis to create data product data model.
+        """Creates a template for a data product model based on the current data model in the session.
+        A data product model is only used for reading of instances.
+        It can be based on any data model, but typically it is based on an enterprise or solution data model.
 
         A data product model is a data model that ONLY maps to containers and do not use implements. This is
         typically used for defining the data in a data product.

cognite_neat-0.117.0/cognite/neat/_version.py

@@ -0,0 +1,2 @@
+__version__ = "0.117.0"
+__engine__ = "^2.0.4"

{cognite_neat-0.116.0 → cognite_neat-0.117.0}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "cognite-neat"
-version = "0.116.0"
+version = "0.117.0"
 readme = "README.md"
 description = "Knowledge graph transformation"
 authors = [
@@ -53,7 +53,7 @@ neat = "cognite.neat._app.main:run"
 python = "^3.10"
 # Core dependencies
 pandas = "*"
-cognite-sdk = "^7.71.2"
+cognite-sdk = "^7.73.4"
 rdflib = "*"
 pydantic = "^2"
 PyYAML = "*"

cognite_neat-0.116.0/cognite/neat/_version.py

@@ -1,2 +0,0 @@
-__version__ = "0.116.0"
-__engine__ = "^2.0.4"