Skip to content

Commit 2fef2de

Browse files
committed
Adding parent project key to experiment container
1 parent 4a99f38 commit 2fef2de

File tree

1 file changed

+2
-1
lines changed
  • metadata-ingestion/src/datahub/ingestion/source

1 file changed

+2
-1
lines changed

metadata-ingestion/src/datahub/ingestion/source/vertexai.py

+2-1
Original file line numberDiff line numberDiff line change
@@ -179,7 +179,7 @@ def get_workunits_internal(self) -> Iterable[MetadataWorkUnit]:
179179
yield from auto_workunit(self._get_training_jobs_mcps())
180180
# Fetch and Ingest Experiments
181181
yield from self._get_experiments_workunits()
182-
# # Fetch and Ingest Experiment Runs
182+
# Fetch and Ingest Experiment Runs
183183
yield from auto_workunit(self._get_experiment_runs_mcps())
184184

185185
def _get_experiments_workunits(self) -> Iterable[MetadataWorkUnit]:
@@ -203,6 +203,7 @@ def _gen_experiment_workunits(
203203
self, experiment: Experiment
204204
) -> Iterable[MetadataWorkUnit]:
205205
yield from gen_containers(
206+
parent_container_key=self._get_project_container(),
206207
container_key=ContainerKeyWithId(
207208
platform=self.platform,
208209
id=self._make_vertexai_experiment_name(experiment.name),

0 commit comments

Comments
 (0)