Fix style + default for scaleProperties

Author: FlorentinD

graphdatascience/procedure_surface/api/catalog/scaler_config.py

@@ -6,5 +6,6 @@ class ScalerConfig(BaseModel):
         description="The type of scaler to use. Can be 'MinMax', 'Mean', 'Max', 'Log', 'StdScore', 'Center'."
     )
     offset: int | float | None = Field(
-        description="The offset to add to the property values before applying the log transformation. Only used when type is 'Log'."
+        default=None,
+        description="The offset to add to the property values before applying the log transformation. Only used when type is 'Log'.",
     )

graphdatascience/procedure_surface/api/node_embedding/__init__.py

@@ -6,8 +6,8 @@
 )
 from graphdatascience.procedure_surface.api.node_embedding.graphsage_endpoints import GraphSageEndpoints
 from graphdatascience.procedure_surface.api.node_embedding.graphsage_predict_endpoints import (
-    GraphSagePredictEndpoints,
     GraphSageMutateResult,
+    GraphSagePredictEndpoints,
     GraphSageWriteResult,
 )
 from graphdatascience.procedure_surface.api.node_embedding.graphsage_train_endpoints import (

graphdatascience/session/__init__.py

@@ -1,11 +1,11 @@
 from .algorithm_category import AlgorithmCategory
 from .aura_api import AuraApiError, SessionStatusError
+from .aura_graph_data_science import AuraGraphDataScience
 from .cloud_location import CloudLocation
 from .dbms_connection_info import DbmsConnectionInfo
 from .gds_sessions import AuraAPICredentials, GdsSessions
 from .session_info import SessionInfo
 from .session_sizes import SessionMemory, SessionMemoryValue
-from .aura_graph_data_science import AuraGraphDataScience
 
 __all__ = [
     "GdsSessions",

graphdatascience/session/session_v2_endpoints.py

@@ -44,10 +44,10 @@
 from graphdatascience.procedure_surface.api.pathfinding.steiner_tree_endpoints import SteinerTreeEndpoints
 from graphdatascience.procedure_surface.api.similarity.knn_endpoints import KnnEndpoints
 from graphdatascience.procedure_surface.api.similarity.node_similarity_endpoints import NodeSimilarityEndpoints
+from graphdatascience.procedure_surface.arrow.catalog.catalog_arrow_endpoints import CatalogArrowEndpoints
 from graphdatascience.procedure_surface.arrow.catalog.scale_properties_arrow_endpoints import (
     ScalePropertiesArrowEndpoints,
 )
-from graphdatascience.procedure_surface.arrow.catalog.catalog_arrow_endpoints import CatalogArrowEndpoints
 from graphdatascience.procedure_surface.arrow.centrality.articlerank_arrow_endpoints import ArticleRankArrowEndpoints
 from graphdatascience.procedure_surface.arrow.centrality.articulationpoints_arrow_endpoints import (
     ArticulationPointsArrowEndpoints,

graphdatascience/tests/integrationV2/procedure_surface/arrow/graph_creation_helper.py

@@ -6,8 +6,8 @@
 from graphdatascience.arrow_client.v2.data_mapper_utils import deserialize_single
 from graphdatascience.arrow_client.v2.job_client import JobClient
 from graphdatascience.procedure_surface.api.catalog.graph_api import GraphV2
-from graphdatascience.procedure_surface.arrow.catalog.graph_backend_arrow import get_graph
 from graphdatascience.procedure_surface.arrow.catalog.catalog_arrow_endpoints import CatalogArrowEndpoints
+from graphdatascience.procedure_surface.arrow.catalog.graph_backend_arrow import get_graph
 
 
 @contextmanager