Training Dataset set Id from backend after creation (#106)

logicalclocks · Sep 28, 2020 · 9987a1f · 9987a1f
1 parent 7f938f1
commit 9987a1f
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/java/src/main/java/com/logicalclocks/hsfs/engine/TrainingDatasetEngine.java b/java/src/main/java/com/logicalclocks/hsfs/engine/TrainingDatasetEngine.java
@@ -67,6 +67,7 @@ public void save(TrainingDataset trainingDataset, Dataset<Row> dataset,
     // Update the original object - Hopsworks returns the full location and incremented version
     trainingDataset.setLocation(apiTD.getLocation());
     trainingDataset.setVersion(apiTD.getVersion());
+    trainingDataset.setId(apiTD.getId());
 
     // Build write options map
     Map<String, String> writeOptions =