Skip to content

Commit 6deecbe

Browse files
committed
use trainingDataset version not id
1 parent 4bc0197 commit 6deecbe

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

hopsworks-api/src/main/java/io/hops/hopsworks/api/provenance/TrainingDatasetProvenanceResource.java

+2-2
Original file line numberDiff line numberDiff line change
@@ -149,7 +149,7 @@ public Response getLinks(
149149
resourceRequest.setExpansions(explicitProvenanceExpansionBeanParam.getResources());
150150
TrainingDataset td
151151
= trainingDatasetController.getTrainingDatasetByFeatureViewAndVersion(getFeatureView(project),
152-
getTrainingDatasetId());
152+
getTrainingDatasetVersion());
153153
ProvExplicitLink<TrainingDataset> provenance
154154
= provCtrl.trainingDatasetLinks(project, td, pagination.getUpstreamLvls(), pagination.getDownstreamLvls());
155155
ProvExplicitLinkDTO<?> result = linksBuilder.build(uriInfo, resourceRequest, project, user, provenance);
@@ -172,7 +172,7 @@ public Response status(@BeanParam ProvUsageBeanParams params,
172172
Project project = getProject();
173173
FeatureView featureView = getFeatureView(project);
174174
TrainingDataset trainingDataset
175-
= trainingDatasetController.getTrainingDatasetByFeatureViewAndVersion(featureView, getTrainingDatasetId());
175+
= trainingDatasetController.getTrainingDatasetByFeatureViewAndVersion(featureView, getTrainingDatasetVersion());
176176
String tdProvenanceId = trainingDataset.getName() + "_" + trainingDataset.getVersion();
177177
ProvArtifactUsageParentDTO status =
178178
usageBuilder.buildAccessible(uriInfo, user, getTrainingDatasetPath(project, featureView),

0 commit comments

Comments
 (0)