Skip to content

Commit

Permalink
[pre-commit.ci] auto fixes from pre-commit.com hooks
Browse files Browse the repository at this point in the history
for more information, see https://pre-commit.ci
  • Loading branch information
pre-commit-ci[bot] authored and andrii-i committed May 30, 2024
1 parent 2969089 commit 9c3a0f9
Showing 1 changed file with 3 additions and 1 deletion.
4 changes: 3 additions & 1 deletion jupyter_scheduler/scheduler.py
Original file line number Diff line number Diff line change
Expand Up @@ -512,7 +512,9 @@ def create_job(self, model: CreateJob) -> str:
input_file_path = os.path.join(self.root_dir, model.input_uri)
mlflow.log_artifact(input_file_path, "input")

mlflow_run = mlflow_client.create_run(experiment_id=experiment_id, run_name=model.input_filename)
mlflow_run = mlflow_client.create_run(
experiment_id=experiment_id, run_name=model.input_filename
)
model.mlflow_run_id = mlflow_run.info.run_id

job = Job(**model.dict(exclude_none=True, exclude={"input_uri"}))
Expand Down

0 comments on commit 9c3a0f9

Please sign in to comment.