Skip to content

Instantly share code, notes, and snippets.

@ywolff
Last active February 15, 2022 01:34
Show Gist options
  • Save ywolff/79db4897ef83a9adba6a8ccff580599e to your computer and use it in GitHub Desktop.
Save ywolff/79db4897ef83a9adba6a8ccff580599e to your computer and use it in GitHub Desktop.
import mlflow
import typer
from src.constants import PROJECT_ROOT_PATH, PROJECT_EXPERIMENT_NAME
def start_pipeline(run_name):
mlflow.set_experiment(PROJECT_EXPERIMENT_NAME)
with mlflow.start_run(run_name=run_name):
print(mlflow.active_run().info.run_id)
mlflow.log_artifact(PROJECT_ROOT_PATH / "dvc.yaml")
if __name__ == "__main__":
typer.run(start_pipeline)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment