MLflow autologging (pymc-labs#921)

radiokosmos · Sep 1, 2024 · 91fd939 · 91fd939
1 parent 6afebc2
commit 91fd939
Show file tree

Hide file tree

Showing 7 changed files with 1,023 additions and 1 deletion.
diff --git a/.github/workflows/test.yml b/.github/workflows/test.yml
@@ -29,6 +29,7 @@ jobs:
         run: pip install pymc==${{ env.OLDEST_PYMC_VERSION }}
       - name: Run tests
         run: |
+          sudo apt-get install graphviz
           pip install -e .[test]
           pytest --cov-report=xml --no-cov-on-fail --durations=50
       - name: Check oldest version of PyMC

diff --git a/.gitignore b/.gitignore
@@ -1,3 +1,7 @@
+# MLflow logging
+mlruns/
+mlruns.db
+
 # InferenceData
 *.nc
 

diff --git a/docs/source/api/index.md b/docs/source/api/index.md
@@ -13,4 +13,5 @@
   model_config
   model_builder
   prior
+  mlflow
 ```
diff --git a/environment.yml b/environment.yml
@@ -44,4 +44,4 @@ dependencies:
 - pytest==7.0.1
 - pytest-cov==3.0.0
 - pytest-mock
-- blas
+- mlflow
-Original file line number
+Diff line change
@@ Expand Up / @@ -13,4 +13,5 @@ @@
       model_config
       model_builder
       prior
+      mlflow
     ```