cognite-neat 0.90.0__tar.gz → 0.90.1__tar.gz

{cognite_neat-0.90.0 → cognite_neat-0.90.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: cognite-neat
-Version: 0.90.0
+Version: 0.90.1
 Summary: Knowledge graph transformation
 Home-page: https://cognite-neat.readthedocs-hosted.com/
 License: Apache-2.0

cognite_neat-0.90.1/cognite/neat/_version.py

@@ -0,0 +1 @@
+__version__ = "0.90.1"

{cognite_neat-0.90.0 → cognite_neat-0.90.1}/cognite/neat/workflows/steps/lib/current/graph_extractor.py

@@ -1,5 +1,4 @@
 import json
-import logging
 from pathlib import Path
 from typing import ClassVar, cast
 
@@ -42,7 +41,7 @@ class GraphFromMockData(Step):
     ]
 
     def run(  # type: ignore[override, syntax]
-        self, rules: MultiRuleData, graph_store: NeatGraph
+        self, rules: MultiRuleData, store: NeatGraph
     ) -> FlowMessage:
         if self.configs is None:
             raise WorkflowStepNotInitializedError(type(self).__name__)
@@ -53,9 +52,6 @@ class GraphFromMockData(Step):
                 step_execution_status=StepExecutionStatus.ABORT_AND_FAIL,
             )
 
-        logging.info(50 * "#")
-        logging.info(50 * "#")
-        logging.info(self.configs["Class count"])
         try:
             class_count = json.loads(self.configs["Class count"]) if self.configs["Class count"] else {}
         except Exception:
@@ -69,9 +65,9 @@ class GraphFromMockData(Step):
             class_count,
         )
 
-        NeatGraph.graph.write(extractor)
+        store.graph.write(extractor)
 
-        return FlowMessage(output_text=f"Instances loaded to the {graph_store.__class__.__name__}")
+        return FlowMessage(output_text=f"Instances loaded to the {store.__class__.__name__}")
 
 
 class GraphFromRdfFile(Step):
@@ -108,12 +104,12 @@ class GraphFromRdfFile(Step):
         ),
     ]
 
-    def run(self, graph_store: NeatGraph) -> FlowMessage:  # type: ignore[override, syntax]
+    def run(self, store: NeatGraph) -> FlowMessage:  # type: ignore[override, syntax]
         if self.configs is None or self.data_store_path is None:
             raise WorkflowStepNotInitializedError(type(self).__name__)
 
         if source_file := self.configs["File path"]:
-            NeatGraph.graph.write(
+            store.graph.write(
                 RdfFileExtractor(  # type: ignore[abstract]
                     filepath=self.data_store_path / Path(source_file),
                     mime_type=self.configs["MIME type"],  # type: ignore[arg-type]

{cognite_neat-0.90.0 → cognite_neat-0.90.1}/cognite/neat/workflows/steps/lib/current/graph_loader.py

@@ -31,7 +31,7 @@ class GraphToRdfFile(Step):
     ]
 
     def run(  # type: ignore[override, syntax]
-        self, graph: NeatGraph
+        self, store: NeatGraph
     ) -> FlowMessage:  # type: ignore[syntax]
         if self.configs is None or self.data_store_path is None:
             raise WorkflowStepNotInitializedError(type(self).__name__)
@@ -39,7 +39,7 @@ class GraphToRdfFile(Step):
         storage_path = self.data_store_path / Path(self.configs["File path"])
         relative_graph_file_path = str(storage_path).split("/data/")[1]
 
-        graph.graph.graph.serialize(str(storage_path), format="turtle")
+        store.graph.graph.serialize(str(storage_path), format="turtle")
 
         output_text = (
             "<p></p>"

{cognite_neat-0.90.0 → cognite_neat-0.90.1}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "cognite-neat"
-version = "0.90.0"
+version = "0.90.1"
 readme = "README.md"
 description = "Knowledge graph transformation"
 authors = [

cognite_neat-0.90.0/cognite/neat/_version.py

@@ -1 +0,0 @@
-__version__ = "0.90.0"