Move Node Embedding algorithm specifications to dedicated module

vnickolov · vnickolov · commit 4363df365abb · 2024-10-24T08:33:41.000+01:00
diff --git a/algorithm-specifications/build.gradle b/algorithm-specifications/build.gradle
@@ -20,6 +20,11 @@ dependencies {
     implementation project(':community-configs')
     implementation project(':community-facade-api')
 
+    implementation project(':node-embeddings-configs')
+    implementation project(':node-embeddings-facade-api')
+
+    implementation project(':path-finding-configs')
+
     // `BaseConfig`
     implementation project(':config-api')
 
@@ -28,6 +33,11 @@ dependencies {
 
     implementation project(':executor')
 
+    // `TrainBaseConfig`
+    implementation project(':ml-algo')
+
+//    implementation project(':model-catalog-api')
+
     // `PageRank` and variants need this
     implementation project(':pregel')
 
diff --git a/algorithm-specifications/src/main/java/org/neo4j/gds/embeddings/fastrp/Constants.java b/algorithm-specifications/src/main/java/org/neo4j/gds/embeddings/fastrp/Constants.java
@@ -19,25 +19,10 @@
  */
 package org.neo4j.gds.embeddings.fastrp;
 
-import org.neo4j.gds.api.properties.nodes.EmptyFloatArrayNodePropertyValues;
-import org.neo4j.gds.api.properties.nodes.NodePropertyValues;
-import org.neo4j.gds.api.properties.nodes.NodePropertyValuesAdapter;
-import org.neo4j.gds.executor.ComputationResult;
-
-final class FastRPCompanion {
+final class Constants {
 
     static final String FASTRP_DESCRIPTION = "Random Projection produces node embeddings via the fastrp algorithm";
 
-    private FastRPCompanion() {}
-
-    static <CONFIG extends FastRPBaseConfig> NodePropertyValues nodeProperties(ComputationResult<FastRP, FastRPResult, CONFIG> computationResult) {
-        return computationResult.result()
-            .map(result -> NodePropertyValuesAdapter.adapt(result.embeddings()))
-            .orElse(EmptyFloatArrayNodePropertyValues.INSTANCE);
-    }
-
-    static NodePropertyValues nodeProperties(FastRPResult result) {
-        return NodePropertyValuesAdapter.adapt(result.embeddings());
-    }
+    private Constants() {}
 
 }
diff --git a/algorithm-specifications/src/main/java/org/neo4j/gds/embeddings/fastrp/FastRPMutateSpec.java b/algorithm-specifications/src/main/java/org/neo4j/gds/embeddings/fastrp/FastRPMutateSpec.java
@@ -31,7 +31,7 @@
 
 import static org.neo4j.gds.executor.ExecutionMode.MUTATE_NODE_PROPERTY;
 
-@GdsCallable(name = "gds.fastRP.mutate", description = FastRPCompanion.FASTRP_DESCRIPTION, executionMode = MUTATE_NODE_PROPERTY)
+@GdsCallable(name = "gds.fastRP.mutate", description = Constants.FASTRP_DESCRIPTION, executionMode = MUTATE_NODE_PROPERTY)
 public class FastRPMutateSpec implements AlgorithmSpec<FastRP, FastRPResult, FastRPMutateConfig, Stream<DefaultNodeEmbeddingMutateResult>, FastRPFactory<FastRPMutateConfig>> {
     @Override
     public String name() {
diff --git a/algorithm-specifications/src/main/java/org/neo4j/gds/embeddings/fastrp/FastRPStatsSpec.java b/algorithm-specifications/src/main/java/org/neo4j/gds/embeddings/fastrp/FastRPStatsSpec.java
@@ -31,7 +31,7 @@
 
 import static org.neo4j.gds.executor.ExecutionMode.STATS;
 
-@GdsCallable(name = "gds.fastRP.stats", description = "Random Projection produces node embeddings via the fastrp algorithm", executionMode = STATS)
+@GdsCallable(name = "gds.fastRP.stats", description = Constants.FASTRP_DESCRIPTION, executionMode = STATS)
 public class FastRPStatsSpec implements AlgorithmSpec<FastRP, FastRPResult, FastRPStatsConfig, Stream<FastRPStatsResult>, FastRPFactory<FastRPStatsConfig>> {
     @Override
     public String name() {
diff --git a/algorithm-specifications/src/main/java/org/neo4j/gds/embeddings/fastrp/FastRPStreamSpec.java b/algorithm-specifications/src/main/java/org/neo4j/gds/embeddings/fastrp/FastRPStreamSpec.java
@@ -31,7 +31,7 @@
 
 import static org.neo4j.gds.executor.ExecutionMode.STREAM;
 
-@GdsCallable(name = "gds.fastRP.stream", description = FastRPCompanion.FASTRP_DESCRIPTION, executionMode = STREAM)
+@GdsCallable(name = "gds.fastRP.stream", description = Constants.FASTRP_DESCRIPTION, executionMode = STREAM)
 public class FastRPStreamSpec implements AlgorithmSpec<FastRP, FastRPResult, FastRPStreamConfig, Stream<FastRPStreamResult>, FastRPFactory<FastRPStreamConfig>> {
     @Override
     public String name() {
diff --git a/algorithm-specifications/src/main/java/org/neo4j/gds/embeddings/fastrp/FastRPWriteSpec.java b/algorithm-specifications/src/main/java/org/neo4j/gds/embeddings/fastrp/FastRPWriteSpec.java
@@ -30,7 +30,7 @@
 
 import java.util.stream.Stream;
 
-import static org.neo4j.gds.embeddings.fastrp.FastRPCompanion.FASTRP_DESCRIPTION;
+import static org.neo4j.gds.embeddings.fastrp.Constants.FASTRP_DESCRIPTION;
 
 @GdsCallable(name = "gds.fastRP.write", description = FASTRP_DESCRIPTION, executionMode = ExecutionMode.WRITE_NODE_PROPERTY)
 public class FastRPWriteSpec implements AlgorithmSpec<FastRP, FastRPResult, FastRPWriteConfig, Stream<DefaultNodeEmbeddingsWriteResult>, FastRPFactory<FastRPWriteConfig>> {
diff --git a/algorithm-specifications/src/main/java/org/neo4j/gds/embeddings/graphsage/Constants.java b/algorithm-specifications/src/main/java/org/neo4j/gds/embeddings/graphsage/Constants.java
@@ -19,23 +19,10 @@
  */
 package org.neo4j.gds.embeddings.graphsage;
 
-import org.neo4j.gds.api.properties.nodes.EmptyDoubleArrayNodePropertyValues;
-import org.neo4j.gds.api.properties.nodes.NodePropertyValues;
-import org.neo4j.gds.api.properties.nodes.NodePropertyValuesAdapter;
-import org.neo4j.gds.embeddings.graphsage.algo.GraphSageResult;
-
-import java.util.Optional;
-
-public final class GraphSageCompanion {
+final class Constants {
 
     static final String GRAPH_SAGE_DESCRIPTION = "The GraphSage algorithm inductively computes embeddings for nodes based on a their features and neighborhoods.";
 
-    private GraphSageCompanion() {}
+    private Constants() {}
 
-    static NodePropertyValues nodePropertyValues(Optional<GraphSageResult> graphSageResult) {
-        return graphSageResult
-            .map(GraphSageResult::embeddings)
-            .map(embeddings -> NodePropertyValuesAdapter.adapt(embeddings))
-            .orElse(EmptyDoubleArrayNodePropertyValues.INSTANCE);
-    }
 }
diff --git a/algorithm-specifications/src/main/java/org/neo4j/gds/embeddings/graphsage/GraphSageConfigurationValidation.java b/algorithm-specifications/src/main/java/org/neo4j/gds/embeddings/graphsage/GraphSageConfigurationValidation.java
diff --git a/algorithm-specifications/src/main/java/org/neo4j/gds/embeddings/graphsage/GraphSageMutateSpec.java b/algorithm-specifications/src/main/java/org/neo4j/gds/embeddings/graphsage/GraphSageMutateSpec.java
@@ -34,7 +34,7 @@
 
 import java.util.stream.Stream;
 
-import static org.neo4j.gds.embeddings.graphsage.GraphSageCompanion.GRAPH_SAGE_DESCRIPTION;
+import static org.neo4j.gds.embeddings.graphsage.Constants.GRAPH_SAGE_DESCRIPTION;
 import static org.neo4j.gds.executor.ExecutionMode.MUTATE_NODE_PROPERTY;
 
 @GdsCallable(name = "gds.beta.graphSage.mutate", description = GRAPH_SAGE_DESCRIPTION, executionMode = MUTATE_NODE_PROPERTY)
diff --git a/algorithm-specifications/src/main/java/org/neo4j/gds/embeddings/graphsage/GraphSageStreamSpec.java b/algorithm-specifications/src/main/java/org/neo4j/gds/embeddings/graphsage/GraphSageStreamSpec.java
@@ -34,7 +34,7 @@
 
 import java.util.stream.Stream;
 
-import static org.neo4j.gds.embeddings.graphsage.GraphSageCompanion.GRAPH_SAGE_DESCRIPTION;
+import static org.neo4j.gds.embeddings.graphsage.Constants.GRAPH_SAGE_DESCRIPTION;
 import static org.neo4j.gds.executor.ExecutionMode.STREAM;
 
 @GdsCallable(name = "gds.beta.graphSage.stream", description = GRAPH_SAGE_DESCRIPTION, executionMode = STREAM)
diff --git a/algorithm-specifications/src/main/java/org/neo4j/gds/embeddings/graphsage/GraphSageTrainSpec.java b/algorithm-specifications/src/main/java/org/neo4j/gds/embeddings/graphsage/GraphSageTrainSpec.java
@@ -33,7 +33,7 @@
 
 import java.util.stream.Stream;
 
-import static org.neo4j.gds.embeddings.graphsage.GraphSageCompanion.GRAPH_SAGE_DESCRIPTION;
+import static org.neo4j.gds.embeddings.graphsage.Constants.GRAPH_SAGE_DESCRIPTION;
 import static org.neo4j.gds.executor.ExecutionMode.TRAIN;
 
 @GdsCallable(name = "gds.beta.graphSage.train", description = GRAPH_SAGE_DESCRIPTION, executionMode = TRAIN)
diff --git a/algorithm-specifications/src/main/java/org/neo4j/gds/embeddings/graphsage/GraphSageWriteSpec.java b/algorithm-specifications/src/main/java/org/neo4j/gds/embeddings/graphsage/GraphSageWriteSpec.java
@@ -34,7 +34,7 @@
 
 import java.util.stream.Stream;
 
-import static org.neo4j.gds.embeddings.graphsage.GraphSageCompanion.GRAPH_SAGE_DESCRIPTION;
+import static org.neo4j.gds.embeddings.graphsage.Constants.GRAPH_SAGE_DESCRIPTION;
 import static org.neo4j.gds.executor.ExecutionMode.WRITE_NODE_PROPERTY;
 
 @GdsCallable(name = "gds.beta.graphSage.write", description = GRAPH_SAGE_DESCRIPTION, executionMode = WRITE_NODE_PROPERTY)
diff --git a/algorithm-specifications/src/main/java/org/neo4j/gds/embeddings/hashgnn/Constants.java b/algorithm-specifications/src/main/java/org/neo4j/gds/embeddings/hashgnn/Constants.java
@@ -19,8 +19,8 @@
  */
 package org.neo4j.gds.embeddings.hashgnn;
 
-final class HashGNNProcCompanion {
+final class Constants {
     static final String HASH_GNN_DESCRIPTION = "HashGNN creates node embeddings by hashing and message passing.";
 
-    private HashGNNProcCompanion() {}
+    private Constants() {}
 }
diff --git a/algorithm-specifications/src/main/java/org/neo4j/gds/embeddings/hashgnn/HashGNNMutateSpec.java b/algorithm-specifications/src/main/java/org/neo4j/gds/embeddings/hashgnn/HashGNNMutateSpec.java
@@ -29,7 +29,7 @@
 
 import java.util.stream.Stream;
 
-import static org.neo4j.gds.embeddings.hashgnn.HashGNNProcCompanion.HASH_GNN_DESCRIPTION;
+import static org.neo4j.gds.embeddings.hashgnn.Constants.HASH_GNN_DESCRIPTION;
 import static org.neo4j.gds.executor.ExecutionMode.MUTATE_NODE_PROPERTY;
 
 @GdsCallable(name = "gds.hashgnn.mutate", aliases = {"gds.beta.hashgnn.mutate"}, description = HASH_GNN_DESCRIPTION, executionMode = MUTATE_NODE_PROPERTY)
diff --git a/algorithm-specifications/src/main/java/org/neo4j/gds/embeddings/hashgnn/HashGNNStreamSpec.java b/algorithm-specifications/src/main/java/org/neo4j/gds/embeddings/hashgnn/HashGNNStreamSpec.java
@@ -29,7 +29,7 @@
 
 import java.util.stream.Stream;
 
-import static org.neo4j.gds.embeddings.hashgnn.HashGNNProcCompanion.HASH_GNN_DESCRIPTION;
+import static org.neo4j.gds.embeddings.hashgnn.Constants.HASH_GNN_DESCRIPTION;
 import static org.neo4j.gds.executor.ExecutionMode.STREAM;
 
 @GdsCallable(name = "gds.hashgnn.stream", aliases = {"gds.beta.hashgnn.stream"}, description = HASH_GNN_DESCRIPTION, executionMode = STREAM)
diff --git a/algorithm-specifications/src/main/java/org/neo4j/gds/embeddings/node2vec/Constants.java b/algorithm-specifications/src/main/java/org/neo4j/gds/embeddings/node2vec/Constants.java
@@ -19,8 +19,8 @@
  */
 package org.neo4j.gds.embeddings.node2vec;
 
-final class Node2VecCompanion {
+final class Constants {
     static final String NODE2VEC_DESCRIPTION = "The Node2Vec algorithm computes embeddings for nodes based on random walks.";
 
-    private Node2VecCompanion() {}
+    private Constants() {}
 }
diff --git a/algorithm-specifications/src/main/java/org/neo4j/gds/embeddings/node2vec/Node2VecMutateSpec.java b/algorithm-specifications/src/main/java/org/neo4j/gds/embeddings/node2vec/Node2VecMutateSpec.java
@@ -34,7 +34,7 @@
 @GdsCallable(
     name = "gds.node2vec.mutate",
     aliases = "gds.beta.node2vec.mutate",
-    description = Node2VecCompanion.NODE2VEC_DESCRIPTION,
+    description = Constants.NODE2VEC_DESCRIPTION,
     executionMode = MUTATE_NODE_PROPERTY
 )
 public class Node2VecMutateSpec implements AlgorithmSpec<Node2Vec, Node2VecResult, Node2VecMutateConfig, Stream<Node2VecMutateResult>, Node2VecAlgorithmFactory<Node2VecMutateConfig>> {
diff --git a/algorithm-specifications/src/main/java/org/neo4j/gds/embeddings/node2vec/Node2VecStreamSpec.java b/algorithm-specifications/src/main/java/org/neo4j/gds/embeddings/node2vec/Node2VecStreamSpec.java
@@ -34,7 +34,7 @@
 @GdsCallable(
     name = "gds.node2vec.stream",
     aliases = "gds.beta.node2vec.stream",
-    description = Node2VecCompanion.NODE2VEC_DESCRIPTION,
+    description = Constants.NODE2VEC_DESCRIPTION,
     executionMode = STREAM
 )
 public class Node2VecStreamSpec implements AlgorithmSpec<Node2Vec, Node2VecResult, Node2VecStreamConfig, Stream<Node2VecStreamResult>, Node2VecAlgorithmFactory<Node2VecStreamConfig>> {
diff --git a/algorithm-specifications/src/main/java/org/neo4j/gds/embeddings/node2vec/Node2VecWriteSpec.java b/algorithm-specifications/src/main/java/org/neo4j/gds/embeddings/node2vec/Node2VecWriteSpec.java
@@ -34,7 +34,7 @@
 @GdsCallable(
     name = "gds.node2vec.write",
     aliases = "gds.beta.node2vec.write",
-    description = Node2VecCompanion.NODE2VEC_DESCRIPTION,
+    description = Constants.NODE2VEC_DESCRIPTION,
     executionMode = WRITE_NODE_PROPERTY
 )
 public class Node2VecWriteSpec implements AlgorithmSpec<Node2Vec, Node2VecResult, Node2VecWriteConfig, Stream<Node2VecWriteResult>, Node2VecAlgorithmFactory<Node2VecWriteConfig>> {
diff --git a/proc/embeddings/build.gradle b/proc/embeddings/build.gradle
@@ -8,52 +8,31 @@ dependencies {
     annotationProcessor project(':annotations')
     annotationProcessor project(':procedure-collector')
 
-    annotationProcessor openGds.immutables.builder
-    annotationProcessor openGds.immutables.value
-
-    // TODO: remove this fella
-    implementation project(':algo')
-
-    implementation project(':algo-common')
-    implementation project(':algorithms-machinery')
-//    implementation project(':algorithms-procedure-facade')
-    implementation project(':annotations')
-    implementation project(':config-api')
-    implementation project(':core')
-    implementation project(':core-write')
-    implementation project(':executor')
-    implementation project(':logging')
-    implementation project(':ml-core')
-    implementation project(':graph-schema-api')
-    implementation project(':model-catalog-api')
-
-//    implementation project(':opengds-procedure-facade')
     implementation project(':procedures-facade-api')
-
-    implementation project(':progress-tracking')
     implementation project(':proc-common')
-    implementation project(':string-formatting')
-    implementation project(':transaction')
-    implementation project(':ml-algo')
-
-    compileOnly openGds.immutables.value.annotations
-    compileOnly openGds.jetbrains.annotations
-    neodeps().each {
-        compileOnly(group: 'org.neo4j', name: it, version: ver.'neo4j') {
-            transitive = false
-        }
-    }
-
-    testAnnotationProcessor project(':annotations')
-
-    testCompileOnly openGds.immutables.builder
-
-    testImplementation project(':native-projection')
-    testImplementation project(':open-model-catalog')
-    testImplementation project(':proc-test')
-    testImplementation project(':licensing')
-    testImplementation project(':open-licensing')
-    testImplementation project(':test-utils')
-
-    testImplementation project(':opengds-extension')
+
+    compileOnly group: 'org.neo4j', name: 'neo4j-procedure-api', version: ver.neo4j
+
+    integrationTestAnnotationProcessor project(':annotations')
+
+    integrationTestCompileOnly openGds.immutables.builder
+
+    integrationTestImplementation project(':algo')
+    integrationTestImplementation project(':annotations')
+    integrationTestImplementation project(':config-api')
+    integrationTestImplementation project(':core')
+    integrationTestImplementation project(':graph-projection-api')
+    integrationTestImplementation project(':graph-schema-api')
+    integrationTestImplementation project(':licensing')
+    integrationTestImplementation project(':ml-algo')
+    integrationTestImplementation project(':ml-core')
+    integrationTestImplementation project(':native-projection')
+    integrationTestImplementation project(':open-model-catalog')
+    integrationTestImplementation project(':open-licensing')
+    integrationTestImplementation project(':proc-test')
+    integrationTestImplementation project(':progress-tracking')
+    integrationTestImplementation project(':string-formatting')
+    integrationTestImplementation project(':test-utils')
+
+    integrationTestImplementation project(':opengds-extension')
 }
diff --git a/proc/embeddings/src/main/java/org/neo4j/gds/embeddings/fastrp/Constants.java b/proc/embeddings/src/main/java/org/neo4j/gds/embeddings/fastrp/Constants.java
@@ -0,0 +1,28 @@
+/*
+ * Copyright (c) "Neo4j"
+ * Neo4j Sweden AB [http://neo4j.com]
+ *
+ * This file is part of Neo4j.
+ *
+ * Neo4j is free software: you can redistribute it and/or modify
+ * it under the terms of the GNU General Public License as published by
+ * the Free Software Foundation, either version 3 of the License, or
+ * (at your option) any later version.
+ *
+ * This program is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+ * GNU General Public License for more details.
+ *
+ * You should have received a copy of the GNU General Public License
+ * along with this program.  If not, see <http://www.gnu.org/licenses/>.
+ */
+package org.neo4j.gds.embeddings.fastrp;
+
+final class Constants {
+
+    static final String FASTRP_DESCRIPTION = "Random Projection produces node embeddings via the fastrp algorithm";
+
+    private Constants() {}
+
+}
diff --git a/proc/embeddings/src/main/java/org/neo4j/gds/embeddings/fastrp/FastRPMutateProc.java b/proc/embeddings/src/main/java/org/neo4j/gds/embeddings/fastrp/FastRPMutateProc.java
@@ -30,7 +30,7 @@
 import java.util.Map;
 import java.util.stream.Stream;
 
-import static org.neo4j.gds.embeddings.fastrp.FastRPCompanion.FASTRP_DESCRIPTION;
+import static org.neo4j.gds.embeddings.fastrp.Constants.FASTRP_DESCRIPTION;
 import static org.neo4j.gds.procedures.ProcedureConstants.MEMORY_ESTIMATION_DESCRIPTION;
 import static org.neo4j.procedure.Mode.READ;
 
diff --git a/proc/embeddings/src/main/java/org/neo4j/gds/embeddings/fastrp/FastRPStatsProc.java b/proc/embeddings/src/main/java/org/neo4j/gds/embeddings/fastrp/FastRPStatsProc.java
@@ -30,7 +30,7 @@
 import java.util.Map;
 import java.util.stream.Stream;
 
-import static org.neo4j.gds.embeddings.fastrp.FastRPCompanion.FASTRP_DESCRIPTION;
+import static org.neo4j.gds.embeddings.fastrp.Constants.FASTRP_DESCRIPTION;
 import static org.neo4j.gds.procedures.ProcedureConstants.MEMORY_ESTIMATION_DESCRIPTION;
 import static org.neo4j.procedure.Mode.READ;
 
diff --git a/proc/embeddings/src/main/java/org/neo4j/gds/embeddings/fastrp/FastRPStreamProc.java b/proc/embeddings/src/main/java/org/neo4j/gds/embeddings/fastrp/FastRPStreamProc.java
@@ -30,7 +30,7 @@
 import java.util.Map;
 import java.util.stream.Stream;
 
-import static org.neo4j.gds.embeddings.fastrp.FastRPCompanion.FASTRP_DESCRIPTION;
+import static org.neo4j.gds.embeddings.fastrp.Constants.FASTRP_DESCRIPTION;
 import static org.neo4j.gds.procedures.ProcedureConstants.MEMORY_ESTIMATION_DESCRIPTION;
 import static org.neo4j.procedure.Mode.READ;
 
diff --git a/proc/embeddings/src/main/java/org/neo4j/gds/embeddings/fastrp/FastRPWriteProc.java b/proc/embeddings/src/main/java/org/neo4j/gds/embeddings/fastrp/FastRPWriteProc.java
@@ -30,7 +30,7 @@
 import java.util.Map;
 import java.util.stream.Stream;
 
-import static org.neo4j.gds.embeddings.fastrp.FastRPCompanion.FASTRP_DESCRIPTION;
+import static org.neo4j.gds.embeddings.fastrp.Constants.FASTRP_DESCRIPTION;
 import static org.neo4j.gds.procedures.ProcedureConstants.MEMORY_ESTIMATION_DESCRIPTION;
 import static org.neo4j.procedure.Mode.READ;
 import static org.neo4j.procedure.Mode.WRITE;
diff --git a/proc/embeddings/src/main/java/org/neo4j/gds/embeddings/graphsage/Constants.java b/proc/embeddings/src/main/java/org/neo4j/gds/embeddings/graphsage/Constants.java
@@ -0,0 +1,28 @@
+/*
+ * Copyright (c) "Neo4j"
+ * Neo4j Sweden AB [http://neo4j.com]
+ *
+ * This file is part of Neo4j.
+ *
+ * Neo4j is free software: you can redistribute it and/or modify
+ * it under the terms of the GNU General Public License as published by
+ * the Free Software Foundation, either version 3 of the License, or
+ * (at your option) any later version.
+ *
+ * This program is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+ * GNU General Public License for more details.
+ *
+ * You should have received a copy of the GNU General Public License
+ * along with this program.  If not, see <http://www.gnu.org/licenses/>.
+ */
+package org.neo4j.gds.embeddings.graphsage;
+
+final class Constants {
+
+    static final String GRAPH_SAGE_DESCRIPTION = "The GraphSage algorithm inductively computes embeddings for nodes based on a their features and neighborhoods.";
+
+    private Constants() {}
+
+}
diff --git a/proc/embeddings/src/main/java/org/neo4j/gds/embeddings/graphsage/GraphSageMutateProc.java b/proc/embeddings/src/main/java/org/neo4j/gds/embeddings/graphsage/GraphSageMutateProc.java
@@ -31,7 +31,7 @@
 import java.util.Map;
 import java.util.stream.Stream;
 
-import static org.neo4j.gds.embeddings.graphsage.GraphSageCompanion.GRAPH_SAGE_DESCRIPTION;
+import static org.neo4j.gds.embeddings.graphsage.Constants.GRAPH_SAGE_DESCRIPTION;
 import static org.neo4j.gds.procedures.ProcedureConstants.MEMORY_ESTIMATION_DESCRIPTION;
 import static org.neo4j.procedure.Mode.READ;
 
diff --git a/proc/embeddings/src/main/java/org/neo4j/gds/embeddings/graphsage/GraphSageStreamProc.java b/proc/embeddings/src/main/java/org/neo4j/gds/embeddings/graphsage/GraphSageStreamProc.java
diff --git a/proc/embeddings/src/main/java/org/neo4j/gds/embeddings/graphsage/GraphSageTrainProc.java b/proc/embeddings/src/main/java/org/neo4j/gds/embeddings/graphsage/GraphSageTrainProc.java
diff --git a/proc/embeddings/src/main/java/org/neo4j/gds/embeddings/graphsage/GraphSageWriteProc.java b/proc/embeddings/src/main/java/org/neo4j/gds/embeddings/graphsage/GraphSageWriteProc.java
diff --git a/proc/embeddings/src/main/java/org/neo4j/gds/embeddings/hashgnn/Constants.java b/proc/embeddings/src/main/java/org/neo4j/gds/embeddings/hashgnn/Constants.java
diff --git a/proc/embeddings/src/main/java/org/neo4j/gds/embeddings/hashgnn/HashGNNMutateProc.java b/proc/embeddings/src/main/java/org/neo4j/gds/embeddings/hashgnn/HashGNNMutateProc.java
diff --git a/proc/embeddings/src/main/java/org/neo4j/gds/embeddings/hashgnn/HashGNNStreamProc.java b/proc/embeddings/src/main/java/org/neo4j/gds/embeddings/hashgnn/HashGNNStreamProc.java
diff --git a/proc/embeddings/src/main/java/org/neo4j/gds/embeddings/node2vec/Constants.java b/proc/embeddings/src/main/java/org/neo4j/gds/embeddings/node2vec/Constants.java
diff --git a/proc/embeddings/src/main/java/org/neo4j/gds/embeddings/node2vec/Node2VecMutateProc.java b/proc/embeddings/src/main/java/org/neo4j/gds/embeddings/node2vec/Node2VecMutateProc.java
diff --git a/proc/embeddings/src/main/java/org/neo4j/gds/embeddings/node2vec/Node2VecStreamProc.java b/proc/embeddings/src/main/java/org/neo4j/gds/embeddings/node2vec/Node2VecStreamProc.java
diff --git a/proc/embeddings/src/main/java/org/neo4j/gds/embeddings/node2vec/Node2VecWriteProc.java b/proc/embeddings/src/main/java/org/neo4j/gds/embeddings/node2vec/Node2VecWriteProc.java
