graphdatascience 1.12a1__tar.gz → 1.14__tar.gz

{graphdatascience-1.12a1/graphdatascience.egg-info → graphdatascience-1.14}/PKG-INFO

@@ -1,6 +1,6 @@
-Metadata-Version: 2.1
+Metadata-Version: 2.2
 Name: graphdatascience
-Version: 1.12a1
+Version: 1.14
 Summary: A Python client for the Neo4j Graph Data Science (GDS) library
 Home-page: https://neo4j.com/product/graph-data-science/
 Author: Neo4j
@@ -16,33 +16,46 @@ Classifier: License :: OSI Approved :: Apache Software License
 Classifier: Operating System :: OS Independent
 Classifier: Programming Language :: Python :: 3
 Classifier: Programming Language :: Python :: 3 :: Only
-Classifier: Programming Language :: Python :: 3.8
 Classifier: Programming Language :: Python :: 3.9
 Classifier: Programming Language :: Python :: 3.10
 Classifier: Programming Language :: Python :: 3.11
 Classifier: Programming Language :: Python :: 3.12
+Classifier: Programming Language :: Python :: 3.13
 Classifier: Topic :: Database
 Classifier: Topic :: Scientific/Engineering
 Classifier: Topic :: Software Development
 Classifier: Typing :: Typed
-Requires-Python: >=3.8
+Requires-Python: >=3.9
 Description-Content-Type: text/markdown
 License-File: LICENSE
-Requires-Dist: multimethod<2.0,>=1.0
+Requires-Dist: multimethod<3.0,>=1.0
 Requires-Dist: neo4j<6.0,>=4.4.12
-Requires-Dist: numpy<2.0
+Requires-Dist: numpy<2.3
 Requires-Dist: pandas<3.0,>=1.0
-Requires-Dist: pyarrow<17.0,>=14.0.1
+Requires-Dist: pyarrow<20.0,>=16.0
 Requires-Dist: textdistance<5.0,>=4.0
 Requires-Dist: tqdm<5.0,>=4.0
 Requires-Dist: typing-extensions<5.0,>=4.0
 Requires-Dist: requests
+Requires-Dist: tenacity>=9.0
 Provides-Extra: ogb
 Requires-Dist: ogb<2.0,>=1.0; extra == "ogb"
 Provides-Extra: networkx
 Requires-Dist: networkx<4.0,>=2.0; extra == "networkx"
 Provides-Extra: rust-ext
-Requires-Dist: neo4j-rust-ext<6.0,>=4.4.2; extra == "rust-ext"
+Requires-Dist: neo4j-rust-ext<6.0,>=4.4.12; extra == "rust-ext"
+Dynamic: author
+Dynamic: author-email
+Dynamic: classifier
+Dynamic: description
+Dynamic: description-content-type
+Dynamic: home-page
+Dynamic: license
+Dynamic: project-url
+Dynamic: provides-extra
+Dynamic: requires-dist
+Dynamic: requires-python
+Dynamic: summary
 
 # Neo4j Graph Data Science Client
 

{graphdatascience-1.12a1 → graphdatascience-1.14}/graphdatascience/algo/algo_proc_runner.py

@@ -1,5 +1,5 @@
 from abc import ABC
-from typing import Any, Dict, Tuple
+from typing import Any
 
 from pandas import DataFrame, Series
 
@@ -12,8 +12,9 @@ from ..model.graphsage_model import GraphSageModel
 
 class AlgoProcRunner(IllegalAttrChecker, ABC):
     @graph_type_check
-    def _run_procedure(self, G: Graph, config: Dict[str, Any], with_logging: bool = True) -> DataFrame:
+    def _run_procedure(self, G: Graph, config: dict[str, Any], with_logging: bool = True) -> DataFrame:
         params = CallParameters(graph_name=G.name(), config=config)
+        params.ensure_job_id_in_config()
 
         return self._query_runner.call_procedure(endpoint=self._namespace, params=params, logging=with_logging)
 
@@ -35,7 +36,7 @@ class StandardModeRunner(AlgoProcRunner):
 
 class GraphSageRunner(AlgoProcRunner):
     @graph_type_check
-    def __call__(self, G: Graph, **config: Any) -> Tuple[GraphSageModel, "Series[Any]"]:
+    def __call__(self, G: Graph, **config: Any) -> tuple[GraphSageModel, "Series[Any]"]:
         result = self._run_procedure(G, config).squeeze()
         model_name = result["modelInfo"]["modelName"]
 

graphdatascience-1.14/graphdatascience/call_parameters.py

@@ -0,0 +1,40 @@
+from collections import OrderedDict
+from typing import Any, Optional
+from uuid import uuid4
+
+
+class CallParameters(OrderedDict[str, Any]):
+    # since Python 3.6 also initializing through CallParameters(**kwargs) is order preserving
+
+    def placeholder_str(self) -> str:
+        return ", ".join([f"${k}" for k in self.keys()])
+
+    def get_job_id(self) -> Optional[str]:
+        config = self["config"] if "config" in self else {}
+
+        job_id = None
+        if "jobId" in config:
+            job_id = config["jobId"]
+
+        if "job_id" in config:
+            job_id = config["job_id"]
+
+        if job_id is None:
+            return None
+
+        return str(job_id)
+
+    def ensure_job_id_in_config(self) -> str:
+        """
+        Ensure that a job id is present in the `config` parameter. If not, generate a new one.
+        This enables the client to check on the progress later on.
+        """
+        config = self.get("config")
+
+        assert config is not None, "config is not set in the parameters. This method should not be called."
+
+        job_id = self.get_job_id()
+        if job_id is None:
+            job_id = str(uuid4())
+            config["jobId"] = job_id
+        return job_id

{graphdatascience-1.12a1 → graphdatascience-1.14}/graphdatascience/error/endpoint_suggester.py

@@ -1,11 +1,9 @@
-from typing import List
-
 import textdistance
 
 from ..ignored_server_endpoints import IGNORED_SERVER_ENDPOINTS
 
 
-def generate_suggestive_error_message(requested_endpoint: str, all_endpoints: List[str]) -> str:
+def generate_suggestive_error_message(requested_endpoint: str, all_endpoints: list[str]) -> str:
     MIN_SIMILARITY_FOR_SUGGESTION = 0.9
 
     closest_endpoint = None

{graphdatascience-1.12a1 → graphdatascience-1.14}/graphdatascience/graph/base_graph_proc_runner.py

@@ -1,8 +1,9 @@
+from __future__ import annotations
+
 import os
 import pathlib
-import sys
 import warnings
-from typing import Any, Dict, List, Optional, Union
+from typing import Any, List, Optional, Union
 
 import pandas as pd
 from multimethod import multimethod
@@ -32,7 +33,6 @@ from .graph_sample_runner import GraphSampleRunner
 from .graph_type_check import (
     from_graph_type_check,
     graph_type_check,
-    graph_type_check_optional,
 )
 from .ogb_loader import OGBLLoader, OGBNLoader
 
@@ -53,28 +53,22 @@ class BaseGraphProcRunner(UncallableNamespace, IllegalAttrChecker):
 
     @staticmethod
     def _path(package: str, resource: str) -> pathlib.Path:
-        if sys.version_info >= (3, 9):
-            from importlib.resources import files
-
-            # files() returns a Traversable, but usages require a Path object
-            return pathlib.Path(str(files(package) / resource))
-        else:
-            from importlib.resources import path
+        from importlib.resources import files
 
-            # we dont want to use a context manager here, so we need to call __enter__ manually
-            return path(package, resource).__enter__()
+        # files() returns a Traversable, but usages require a Path object
+        return pathlib.Path(str(files(package) / resource))
 
     @client_only_endpoint("gds.graph")
     @compatible_with("construct", min_inclusive=ServerVersion(2, 1, 0))
     def construct(
         self,
         graph_name: str,
-        nodes: Union[DataFrame, List[DataFrame]],
-        relationships: Optional[Union[DataFrame, List[DataFrame]]] = None,
+        nodes: Union[DataFrame, list[DataFrame]],
+        relationships: Optional[Union[DataFrame, list[DataFrame]]] = None,
         concurrency: int = 4,
-        undirected_relationship_types: Optional[List[str]] = None,
+        undirected_relationship_types: Optional[list[str]] = None,
     ) -> Graph:
-        nodes = nodes if isinstance(nodes, List) else [nodes]
+        nodes = nodes if isinstance(nodes, list) else [nodes]
 
         if isinstance(relationships, DataFrame):
             relationships = [relationships]
@@ -244,6 +238,7 @@ class BaseGraphProcRunner(UncallableNamespace, IllegalAttrChecker):
         params = CallParameters(
             graph_name=graph_name, node_count=node_count, average_degree=average_degree, config=config
         )
+        params.ensure_job_id_in_config()
 
         result = self._query_runner.call_procedure(
             endpoint=self._namespace,
@@ -270,6 +265,8 @@ class BaseGraphProcRunner(UncallableNamespace, IllegalAttrChecker):
             relationship_filter=relationship_filter,
             config=config,
         )
+        params.ensure_job_id_in_config()
+
         result = self._query_runner.call_procedure(
             endpoint=self._namespace,
             logging=True,
@@ -299,7 +296,7 @@ class BaseGraphProcRunner(UncallableNamespace, IllegalAttrChecker):
         failIfMissing: bool = False,
         dbName: str = "",
         username: Optional[str] = None,
-    ) -> Optional["Series[Any]"]:
+    ) -> Optional[Series[Any]]:
         self._namespace += ".drop"
 
         if isinstance(graph, Graph):
@@ -322,7 +319,7 @@ class BaseGraphProcRunner(UncallableNamespace, IllegalAttrChecker):
 
         return None
 
-    def exists(self, graph_name: str) -> "Series[Any]":
+    def exists(self, graph_name: str) -> Series[Any]:
         self._namespace += ".exists"
         result = self._query_runner.call_procedure(
             endpoint=self._namespace,
@@ -331,13 +328,17 @@ class BaseGraphProcRunner(UncallableNamespace, IllegalAttrChecker):
 
         return result.squeeze()  # type: ignore
 
-    @graph_type_check_optional
-    def list(self, G: Optional[Graph] = None) -> DataFrame:
+    def list(self, G: Optional[Union[Graph, str]] = None) -> DataFrame:
         self._namespace += ".list"
 
+        if isinstance(G, Graph):
+            graph_name = G.name()
+        elif isinstance(G, str):
+            graph_name = G
+
         params = CallParameters()
         if G:
-            params["graph_name"] = G.name()
+            params["graph_name"] = graph_name
 
         return self._query_runner.call_procedure(
             endpoint=self._namespace,
@@ -365,7 +366,7 @@ class BaseGraphProcRunner(UncallableNamespace, IllegalAttrChecker):
         G: Graph,
         properties: Strings,
         entities: Strings,
-        config: Dict[str, Any],
+        config: dict[str, Any],
     ) -> DataFrame:
         params = CallParameters(
             graph_name=G.name(),

{graphdatascience-1.12a1 → graphdatascience-1.14}/graphdatascience/graph/graph_alpha_proc_runner.py

@@ -1,5 +1,5 @@
 import os
-from typing import List, Optional, Union
+from typing import Optional, Union
 
 from pandas import DataFrame
 
@@ -36,13 +36,13 @@ class GraphAlphaProcRunner(UncallableNamespace, IllegalAttrChecker):
     def construct(
         self,
         graph_name: str,
-        nodes: Union[DataFrame, List[DataFrame]],
-        relationships: Union[DataFrame, List[DataFrame]],
+        nodes: Union[DataFrame, list[DataFrame]],
+        relationships: Union[DataFrame, list[DataFrame]],
         concurrency: int = 4,
-        undirected_relationship_types: Optional[List[str]] = None,
+        undirected_relationship_types: Optional[list[str]] = None,
     ) -> Graph:
-        nodes = nodes if isinstance(nodes, List) else [nodes]
-        relationships = relationships if isinstance(relationships, List) else [relationships]
+        nodes = nodes if isinstance(nodes, list) else [nodes]
+        relationships = relationships if isinstance(relationships, list) else [relationships]
 
         errors = []
 

{graphdatascience-1.12a1 → graphdatascience-1.14}/graphdatascience/graph/graph_beta_proc_runner.py

@@ -1,4 +1,4 @@
-from typing import Any, List, Union
+from typing import Any, Union
 
 from ..call_parameters import CallParameters
 from ..error.illegal_attr_checker import IllegalAttrChecker
@@ -9,7 +9,7 @@ from .graph_export_runner import GraphExportCsvEndpoints
 from .graph_object import Graph
 from .graph_project_runner import GraphProjectBetaRunner
 
-Strings = Union[str, List[str]]
+Strings = Union[str, list[str]]
 
 
 class GraphBetaProcRunner(UncallableNamespace, IllegalAttrChecker):

{graphdatascience-1.12a1 → graphdatascience-1.14}/graphdatascience/graph/graph_cypher_runner.py

@@ -2,7 +2,7 @@ from __future__ import annotations
 
 import re
 from itertools import chain, zip_longest
-from typing import Any, Dict, Optional
+from typing import Any, Optional
 
 from pandas import Series
 
@@ -45,7 +45,7 @@ class GraphCypherRunner(CallerBase):
 
         GraphCypherRunner._verify_query_ends_with_return_clause(self._namespace, query)
 
-        result: Optional[Dict[str, Any]] = self._query_runner.run_cypher(query, params, database, False).squeeze()
+        result: Optional[dict[str, Any]] = self._query_runner.run_cypher(query, params, database, False).squeeze()
 
         if not result:
             raise ValueError("Projected graph cannot be empty.")

{graphdatascience-1.12a1 → graphdatascience-1.14}/graphdatascience/graph/graph_entity_ops_runner.py

@@ -1,5 +1,5 @@
 from functools import reduce
-from typing import Any, Dict, List, Type, Union
+from typing import Any, Type, Union
 from warnings import filterwarnings
 
 import pandas as pd
@@ -18,7 +18,7 @@ from ..utils.util_proc_runner import UtilProcRunner
 from .graph_object import Graph
 from .graph_type_check import graph_type_check
 
-Strings = Union[str, List[str]]
+Strings = Union[str, list[str]]
 
 
 class TopologyDataFrame(DataFrame):
@@ -26,7 +26,7 @@ class TopologyDataFrame(DataFrame):
     def _constructor(self) -> "Type[TopologyDataFrame]":
         return TopologyDataFrame
 
-    def by_rel_type(self) -> Dict[str, List[List[int]]]:
+    def by_rel_type(self) -> dict[str, list[list[int]]]:
         # Pandas 2.2.0 deprecated an internal API used by DF.take(indices)
         filterwarnings(
             "ignore",
@@ -38,7 +38,7 @@ class TopologyDataFrame(DataFrame):
 
         output = {}
         for rel_type, indices in gb.groups.items():
-            one_rel_df = self.take(indices)
+            one_rel_df = self.take(list(indices))
             output[str(rel_type)] = [list(one_rel_df["sourceNodeId"]), list(one_rel_df["targetNodeId"])]
 
         return output
@@ -55,7 +55,7 @@ class GraphEntityOpsBaseRunner(UncallableNamespace, IllegalAttrChecker):
         G: Graph,
         properties: Strings,
         entities: Strings,
-        config: Dict[str, Any],
+        config: dict[str, Any],
     ) -> DataFrame:
         params = CallParameters(
             graph_name=G.name(),
@@ -78,11 +78,16 @@ class GraphNodePropertyRunner(GraphEntityOpsBaseRunner):
         G: Graph,
         node_property: str,
         node_labels: Strings = ["*"],
-        db_node_properties: List[str] = [],
+        db_node_properties: list[str] = [],
         **config: Any,
     ) -> DataFrame:
         self._namespace += ".stream"
 
+        if len(set(db_node_properties)) != len(db_node_properties):
+            raise ValueError(
+                f"The provided db_node_properties contain duplicate property names: `{db_node_properties}`."
+            )
+
         result = self._handle_properties(G, node_property, node_labels, config)
 
         return GraphNodePropertiesRunner._process_result(
@@ -96,14 +101,25 @@ class GraphNodePropertiesRunner(GraphEntityOpsBaseRunner):
     def stream(
         self,
         G: Graph,
-        node_properties: List[str],
+        node_properties: Union[str, list[str]],
         node_labels: Strings = ["*"],
         separate_property_columns: bool = False,
-        db_node_properties: List[str] = [],
+        db_node_properties: list[str] = [],
         **config: Any,
     ) -> DataFrame:
         self._namespace += ".stream"
 
+        if isinstance(node_properties, str):
+            node_properties = [node_properties]
+
+        # find if list contain duplicates
+        if len(set(node_properties)) != len(node_properties):
+            raise ValueError(f"The provided node_properties contain duplicate property names: `{node_properties}`.")
+        if len(set(db_node_properties)) != len(db_node_properties):
+            raise ValueError(
+                f"The provided db_node_properties contain duplicate property names: `{db_node_properties}`."
+            )
+
         result = self._handle_properties(G, node_properties, node_labels, config)
 
         return GraphNodePropertiesRunner._process_result(
@@ -113,11 +129,11 @@ class GraphNodePropertiesRunner(GraphEntityOpsBaseRunner):
     @staticmethod
     def _process_result(
         query_runner: QueryRunner,
-        node_properties: List[str],
+        node_properties: list[str],
         separate_property_columns: bool,
-        db_node_properties: List[str],
+        db_node_properties: list[str],
         result: DataFrame,
-        config: Dict[str, Any],
+        config: dict[str, Any],
     ) -> DataFrame:
         # new format was requested, but the query was run via Cypher
         if separate_property_columns and "propertyValue" in result.keys():
@@ -125,7 +141,11 @@ class GraphNodePropertiesRunner(GraphEntityOpsBaseRunner):
             if "listNodeLabels" in config.keys():
                 # nodeLabels cannot be an index column of the pivot as its not hashable
                 # so we need to manually join it back in
-                labels_df = result[["nodeId", "nodeLabels"]].set_index("nodeId")
+                labels_df = result[["nodeId", "nodeLabels"]]
+                # drop duplicates as for each property we would have one row for each nodeId
+                labels_df = labels_df.drop_duplicates(ignore_index=False, subset=["nodeId"])
+                labels_df.set_index("nodeId", inplace=True)
+
                 wide_result = wide_result.join(labels_df, on="nodeId")
             result = wide_result.reset_index()
             result.columns.name = None
@@ -161,7 +181,7 @@ class GraphNodePropertiesRunner(GraphEntityOpsBaseRunner):
         return result
 
     @staticmethod
-    def _build_query(db_node_properties: List[str]) -> str:
+    def _build_query(db_node_properties: list[str]) -> str:
         query_prefix = "MATCH (n) WHERE id(n) IN $ids RETURN id(n) AS nodeId"
 
         def add_property(query: str, prop: str) -> str:
@@ -176,7 +196,7 @@ class GraphNodePropertiesRunner(GraphEntityOpsBaseRunner):
 
     @compatible_with("drop", min_inclusive=ServerVersion(2, 2, 0))
     @graph_type_check
-    def drop(self, G: Graph, node_properties: List[str], **config: Any) -> "Series[Any]":
+    def drop(self, G: Graph, node_properties: list[str], **config: Any) -> "Series[Any]":
         self._namespace += ".drop"
         params = CallParameters(
             graph_name=G.name(),
@@ -205,7 +225,7 @@ class GraphRelationshipPropertiesRunner(GraphEntityOpsBaseRunner):
     def stream(
         self,
         G: Graph,
-        relationship_properties: List[str],
+        relationship_properties: list[str],
         relationship_types: Strings = ["*"],
         separate_property_columns: bool = False,
         **config: Any,
@@ -239,7 +259,7 @@ class GraphRelationshipPropertiesRunner(GraphEntityOpsBaseRunner):
         self,
         G: Graph,
         relationship_type: str,
-        relationship_properties: List[str],
+        relationship_properties: list[str],
         **config: Any,
     ) -> "Series[Any]":
         self._namespace += ".write"
@@ -319,7 +339,7 @@ class GraphRelationshipsRunner(GraphEntityOpsBaseRunner):
 
     @compatible_with("stream", min_inclusive=ServerVersion(2, 5, 0))
     @graph_type_check
-    def stream(self, G: Graph, relationship_types: List[str] = ["*"], **config: Any) -> TopologyDataFrame:
+    def stream(self, G: Graph, relationship_types: list[str] = ["*"], **config: Any) -> TopologyDataFrame:
         self._namespace += ".stream"
         params = CallParameters(graph_name=G.name(), relationship_types=relationship_types, config=config)
         result = self._query_runner.call_procedure(endpoint=self._namespace, params=params)
@@ -336,7 +356,7 @@ class GraphRelationshipsRunner(GraphEntityOpsBaseRunner):
 class GraphRelationshipsBetaRunner(GraphEntityOpsBaseRunner):
     @compatible_with("stream", min_inclusive=ServerVersion(2, 2, 0))
     @graph_type_check
-    def stream(self, G: Graph, relationship_types: List[str] = ["*"], **config: Any) -> TopologyDataFrame:
+    def stream(self, G: Graph, relationship_types: list[str] = ["*"], **config: Any) -> TopologyDataFrame:
         self._namespace += ".stream"
         params = CallParameters(graph_name=G.name(), relationship_types=relationship_types, config=config)
 

{graphdatascience-1.12a1 → graphdatascience-1.14}/graphdatascience/graph/graph_export_runner.py

@@ -1,4 +1,4 @@
-from typing import Any, Dict
+from typing import Any
 
 from pandas import Series
 
@@ -15,7 +15,7 @@ class GraphExportCsvRunner(IllegalAttrChecker):
         return self._export_call(G, config)
 
     @graph_type_check
-    def _export_call(self, G: Graph, config: Dict[str, Any]) -> "Series[Any]":
+    def _export_call(self, G: Graph, config: dict[str, Any]) -> "Series[Any]":
         params = CallParameters(graph_name=G.name(), config=config)
         return self._query_runner.call_procedure(endpoint=self._namespace, params=params).squeeze()  # type: ignore
 
@@ -39,7 +39,7 @@ class GraphExportRunner(IllegalAttrChecker):
         return self._export_call(G, config)
 
     @graph_type_check
-    def _export_call(self, G: Graph, config: Dict[str, Any]) -> "Series[Any]":
+    def _export_call(self, G: Graph, config: dict[str, Any]) -> "Series[Any]":
         params = CallParameters(graph_name=G.name(), config=config)
         return self._query_runner.call_procedure(endpoint=self._namespace, params=params).squeeze()  # type: ignore
 

{graphdatascience-1.12a1 → graphdatascience-1.14}/graphdatascience/graph/graph_object.py

@@ -1,7 +1,7 @@
 from __future__ import annotations
 
 from types import TracebackType
-from typing import Any, List, Optional, Type, Union
+from typing import Any, Optional, Type, Union
 
 from pandas import Series
 
@@ -39,7 +39,7 @@ class Graph:
         """
         return self._name
 
-    def _graph_info(self, yields: List[str] = []) -> "Series[Any]":
+    def _graph_info(self, yields: list[str] = []) -> Series[Any]:
         yield_db = "database" in yields
         yields_with_db = yields if yield_db else yields + ["database"]
 
@@ -90,21 +90,21 @@ class Graph:
         """
         return self._graph_info(["relationshipCount"])  # type: ignore
 
-    def node_labels(self) -> List[str]:
+    def node_labels(self) -> list[str]:
         """
         Returns:
             the node labels in the graph
         """
         return list(self._graph_info(["schema"])["nodes"].keys())
 
-    def relationship_types(self) -> List[str]:
+    def relationship_types(self) -> list[str]:
         """
         Returns:
             the relationship types in the graph
         """
         return list(self._graph_info(["schema"])["relationships"].keys())
 
-    def node_properties(self, label: Optional[str] = None) -> Union["Series[str]", List[str]]:
+    def node_properties(self, label: Optional[str] = None) -> Union[Series[str], list[str]]:
         """
         Args:
             label: the node label to get the properties for
@@ -123,7 +123,7 @@ class Graph:
 
         return list(labels_to_props[label].keys())
 
-    def relationship_properties(self, type: Optional[str] = None) -> Union["Series[str]", List[str]]:
+    def relationship_properties(self, type: Optional[str] = None) -> Union[Series[str], list[str]]:
         """
         Args:
             type: the relationship type to get the properties for

{graphdatascience-1.12a1 → graphdatascience-1.14}/graphdatascience/graph/graph_remote_project_runner.py

@@ -1,6 +1,6 @@
 from __future__ import annotations
 
-from typing import List, Optional
+from typing import Optional
 
 from ..call_parameters import CallParameters
 from ..error.illegal_attr_checker import IllegalAttrChecker
@@ -17,10 +17,12 @@ class GraphProjectRemoteRunner(IllegalAttrChecker):
         self,
         graph_name: str,
         query: str,
+        job_id: Optional[str] = None,
         concurrency: int = 4,
-        undirected_relationship_types: Optional[List[str]] = None,
-        inverse_indexed_relationship_types: Optional[List[str]] = None,
+        undirected_relationship_types: Optional[list[str]] = None,
+        inverse_indexed_relationship_types: Optional[list[str]] = None,
         batch_size: Optional[int] = None,
+        logging: bool = True,
     ) -> GraphCreateResult:
         if inverse_indexed_relationship_types is None:
             inverse_indexed_relationship_types = []
@@ -34,6 +36,7 @@ class GraphProjectRemoteRunner(IllegalAttrChecker):
         params = CallParameters(
             graph_name=graph_name,
             query=query,
+            job_id=job_id,
             concurrency=concurrency,
             undirected_relationship_types=undirected_relationship_types,
             inverse_indexed_relationship_types=inverse_indexed_relationship_types,
@@ -43,5 +46,6 @@ class GraphProjectRemoteRunner(IllegalAttrChecker):
         result = self._query_runner.call_procedure(
             endpoint=SessionQueryRunner.GDS_REMOTE_PROJECTION_PROC_NAME,
             params=params,
+            logging=True,
         ).squeeze()
         return GraphCreateResult(Graph(graph_name, self._query_runner), result)

{graphdatascience-1.12a1 → graphdatascience-1.14}/graphdatascience/graph/graph_type_check.py

@@ -1,5 +1,5 @@
 from functools import wraps
-from typing import Any, Callable, Optional, TypeVar, cast
+from typing import Any, Callable, TypeVar, cast
 
 from .graph_object import Graph
 
@@ -20,20 +20,6 @@ def graph_type_check(func: F) -> F:
     return cast(F, wrapper)
 
 
-def graph_type_check_optional(func: F) -> F:
-    @wraps(func)
-    def wrapper(self: Any, G: Optional[Graph] = None, *args: Any, **kwargs: Any) -> Any:
-        if isinstance(G, str):
-            raise TypeError(
-                f"The parameter 'G' takes a `Graph` object, but received string '{G}'. "
-                "To resolve a graph name string into a `Graph` object, please use `gds.graph.get`"
-            )
-
-        return func(self, G, *args, **kwargs)
-
-    return cast(F, wrapper)
-
-
 def from_graph_type_check(func: F) -> F:
     @wraps(func)
     def wrapper(self: Any, graph_name: str, from_G: Graph, *args: Any, **kwargs: Any) -> Any: