Skip to content

Commit ea6e34a

Browse files
committed
Fix node schema construction in Arrow GDS import
1 parent 95b506e commit ea6e34a

File tree

1 file changed

+10
-4
lines changed
  • core/src/main/java/org/neo4j/gds/core/loading

1 file changed

+10
-4
lines changed

core/src/main/java/org/neo4j/gds/core/loading/Nodes.java

Lines changed: 10 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -95,9 +95,14 @@ static Nodes of(
9595
return ImmutableNodes.of(nodeSchema, idMap, nodePropertyStoreBuilder.build());
9696
}
9797

98-
static Nodes of(NodeSchema nodeSchema, IdMap idMap, Map<PropertyMapping, NodePropertyValues> properties, PropertyState propertyState) {
99-
NodePropertyStore.Builder builder = NodePropertyStore.builder();
100-
properties.forEach((mapping, nodeProperties) -> builder.putProperty(
98+
static Nodes of(
99+
NodeSchema nodeSchema,
100+
IdMap idMap,
101+
Map<PropertyMapping, NodePropertyValues> properties,
102+
PropertyState propertyState
103+
) {
104+
var propertyStoreBuilder = NodePropertyStore.builder();
105+
properties.forEach((mapping, nodeProperties) -> propertyStoreBuilder.putProperty(
101106
mapping.propertyKey(),
102107
NodeProperty.of(
103108
mapping.propertyKey(),
@@ -108,6 +113,7 @@ static Nodes of(NodeSchema nodeSchema, IdMap idMap, Map<PropertyMapping, NodePro
108113
: nodeProperties.valueType().fallbackValue()
109114
)
110115
));
111-
return ImmutableNodes.of(nodeSchema, idMap, builder.build());
116+
117+
return ImmutableNodes.of(nodeSchema, idMap, propertyStoreBuilder.build());
112118
}
113119
}

0 commit comments

Comments
 (0)