Fix typing + signature mismatch

Author: FlorentinD

graphdatascience/procedure_surface/api/centrality/celf_endpoints.py

@@ -191,7 +191,7 @@ def write(
         random_seed: int | None = None,
         relationship_types: list[str] = ALL_TYPES,
         node_labels: list[str] = ALL_LABELS,
-        sudo: bool | None = None,
+        sudo: bool = False,
         log_progress: bool = True,
         username: str | None = None,
         concurrency: int | None = None,

graphdatascience/tests/integrationV2/procedure_surface/session/test_session_api_spec_coverage.py

@@ -102,7 +102,7 @@
     ],
 }
 
-ADJUSTED_PARAM_DEFAULT_VALUES = {
+ADJUSTED_PARAM_DEFAULT_VALUES: dict[str, dict[str, str | None]] = {
     ".*": {
         "concurrency": None,  # default value differs for Aura Graph Analytics compared to plugin (spec is off)
         "job_id": None,  # default value in spec is `random id`
@@ -248,7 +248,7 @@ def verify_configuration_fields(callable_object: MethodType, endpoint_spec: Endp
         )
 
     # validate default values match
-    default_adjustments: dict[str, str] = {}
+    default_adjustments: dict[str, str | None] = {}
     for endpoint_pattern, adjustments in ADJUSTED_PARAM_DEFAULT_VALUES.items():
         if re.match(endpoint_pattern, py_endpoint):
             default_adjustments.update(adjustments)

graphdatascience/tests/unit/procedure_surface/cypher/community/test_unit_k1coloring_cypher_endpoints.py

@@ -7,6 +7,7 @@
     K1ColoringStatsResult,
     K1ColoringWriteResult,
 )
+from graphdatascience.procedure_surface.api.default_values import ALL_LABELS, ALL_TYPES
 from graphdatascience.procedure_surface.cypher.community.k1coloring_cypher_endpoints import K1ColoringCypherEndpoints
 from graphdatascience.tests.unit.conftest import DEFAULT_SERVER_VERSION, CollectingQueryRunner
 from graphdatascience.tests.unit.procedure_surface.cypher.conftest import estimate_mock_result
@@ -317,7 +318,12 @@ def test_estimate_with_graph_name(graph: GraphV2) -> None:
     assert "gds.k1coloring.stats.estimate" in query_runner.queries[0]
     params = query_runner.params[0]
     assert params["graphNameOrConfiguration"] == "test_graph"
-    assert params["algoConfig"] == {}
+    assert params["algoConfig"] == {
+        "batchSize": 10000,
+        "maxIterations": 10,
+        "nodeLabels": ALL_LABELS,
+        "relationshipTypes": ALL_TYPES,
+    }
 
 
 def test_estimate_with_projection_config() -> None:
@@ -331,4 +337,9 @@ def test_estimate_with_projection_config() -> None:
     assert "gds.k1coloring.stats.estimate" in query_runner.queries[0]
     params = query_runner.params[0]
     assert params["graphNameOrConfiguration"] == {"foo": "bar"}
-    assert params["algoConfig"] == {}
+    assert params["algoConfig"] == {
+        "batchSize": 10000,
+        "maxIterations": 10,
+        "nodeLabels": ALL_LABELS,
+        "relationshipTypes": ALL_TYPES,
+    }