use trainingDataset version not id

logicalclocks · May 15, 2024 · 6deecbe · 6deecbe
1 parent 4bc0197
commit 6deecbe
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/...api/src/main/java/io/hops/hopsworks/api/provenance/TrainingDatasetProvenanceResource.java b/...api/src/main/java/io/hops/hopsworks/api/provenance/TrainingDatasetProvenanceResource.java
@@ -149,7 +149,7 @@ public Response getLinks(
     resourceRequest.setExpansions(explicitProvenanceExpansionBeanParam.getResources());
     TrainingDataset td
       = trainingDatasetController.getTrainingDatasetByFeatureViewAndVersion(getFeatureView(project),
-      getTrainingDatasetId());
+      getTrainingDatasetVersion());
     ProvExplicitLink<TrainingDataset> provenance
         = provCtrl.trainingDatasetLinks(project, td, pagination.getUpstreamLvls(), pagination.getDownstreamLvls());
     ProvExplicitLinkDTO<?> result = linksBuilder.build(uriInfo, resourceRequest, project, user, provenance);
@@ -172,7 +172,7 @@ public Response status(@BeanParam ProvUsageBeanParams params,
     Project project = getProject();
     FeatureView featureView = getFeatureView(project);
     TrainingDataset trainingDataset
-        = trainingDatasetController.getTrainingDatasetByFeatureViewAndVersion(featureView, getTrainingDatasetId());
+        = trainingDatasetController.getTrainingDatasetByFeatureViewAndVersion(featureView, getTrainingDatasetVersion());
     String tdProvenanceId = trainingDataset.getName() + "_" + trainingDataset.getVersion();
     ProvArtifactUsageParentDTO status =
       usageBuilder.buildAccessible(uriInfo, user, getTrainingDatasetPath(project, featureView),