MLflow autologging

.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

.gitignore

@@ -1,3 +1,7 @@
+# MLflow logging
+mlruns/
+mlruns.db
+
 # InferenceData
 *.nc
 

docs/source/api/index.md

@@ -13,4 +13,5 @@
   model_config
   model_builder
   prior
+  mlflow
 ```

environment.yml

@@ -44,3 +44,4 @@ dependencies:
 - pytest==7.0.1
 - pytest-cov==3.0.0
 - pytest-mock
+- mlflow