diff --git a/test.py b/test.py index 6408136..75a0071 100644 --- a/test.py +++ b/test.py @@ -15,6 +15,8 @@ from sklearn.linear_model import ElasticNet import mlflow import mlflow.sklearn +mlflow.set_tracking_uri(uri="http://127.0.0.1:80") + import logging logging.basicConfig(level=logging.WARN) logger = logging.getLogger(__name__) diff --git a/testgit.py b/testgit.py new file mode 100644 index 0000000..c5a291d --- /dev/null +++ b/testgit.py @@ -0,0 +1,9 @@ +import mlflow + +run = mlflow.get_run("82dc1facd73b4fda886b25005e5db0d8") +git_commit = run.data.tags.get("mlflow.source.git.commit") +git_repo_url = run.data.tags.get("mlflow.source.git.repoURL") + +print(f"Git Commit: {git_commit}") +print(f"Git Repo URL: {git_repo_url}") +