Skip to content

Commit

Permalink
Update runner to use KfP v2 API
Browse files Browse the repository at this point in the history
  • Loading branch information
RobbeSneyders committed Oct 9, 2023
1 parent cf179f1 commit 55fc4fe
Showing 1 changed file with 3 additions and 3 deletions.
6 changes: 3 additions & 3 deletions src/fondant/runner.py
Original file line number Diff line number Diff line change
Expand Up @@ -73,18 +73,18 @@ def run(
job_name = self.get_name_from_spec(input_spec) + "_run"
# TODO add logic to see if pipeline exists
runner = self.client.run_pipeline(
experiment_id=experiment.id,
experiment_id=experiment.experiment_id,
job_name=job_name,
pipeline_package_path=input_spec,
)

pipeline_url = f"{self.host}/#/runs/details/{runner.id}"
pipeline_url = f"{self.host}/#/runs/details/{runner.run_id}"
logger.info(f"Pipeline is running at: {pipeline_url}")

def get_name_from_spec(self, input_spec: str):
"""Get the name of the pipeline from the spec."""
with open(input_spec) as f:
spec = yaml.safe_load(f)
spec, *_ = yaml.safe_load_all(f)
return spec["pipelineInfo"]["name"]


Expand Down

0 comments on commit 55fc4fe

Please sign in to comment.