diff --git a/requirements.txt b/requirements.txt
index 2cabd50..efb6db4 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -10,6 +10,7 @@ absl-py==0.15.0
     #   tensorflow
 aiobotocore[boto3]==1.3.0
     # via
+    #   aiobotocore
     #   dvc
     #   s3fs
 aiohttp==3.8.1
@@ -29,7 +30,7 @@ astunparse==1.6.3
 async-timeout==4.0.2
     # via aiohttp
 atpublic==2.3
-    # via flufl.lock
+    # via flufl-lock
 attrs==21.4.0
     # via
     #   aiohttp
@@ -74,6 +75,10 @@ configobj==5.0.6
     # via dvc
 configparser==5.2.0
     # via wandb
+contourpy==1.1.1
+    # via matplotlib
+cycler==0.12.1
+    # via matplotlib
 databricks-cli==0.16.2
     # via mlflow
 decorator==5.1.1
@@ -101,15 +106,15 @@ entrypoints==0.3
 executing==0.8.2
     # via icecream
 flask==2.0.2
-    # via
-    #   mlflow
-    #   prometheus-flask-exporter
+    # via mlflow
 flatbuffers==1.12
     # via tensorflow
 flatten-dict==0.4.2
     # via dvc
-flufl.lock==3.2
+flufl-lock==3.2
     # via dvc
+fonttools==4.46.0
+    # via matplotlib
 frozenlist==1.2.0
     # via
     #   aiohttp
@@ -168,7 +173,9 @@ iniconfig==1.1.1
 itsdangerous==2.0.1
     # via flask
 jinja2==3.0.3
-    # via flask
+    # via
+    #   flask
+    #   mlflow
 jmespath==0.10.0
     # via
     #   boto3
@@ -181,6 +188,8 @@ keras==2.7.0
     # via tensorflow
 keras-preprocessing==1.1.2
     # via tensorflow
+kiwisolver==1.4.5
+    # via matplotlib
 libclang==12.0.0
     # via tensorflow
 mailchecker==4.1.7
@@ -188,12 +197,16 @@ mailchecker==4.1.7
 mako==1.1.6
     # via alembic
 markdown==3.3.6
-    # via tensorboard
+    # via
+    #   mlflow
+    #   tensorboard
 markupsafe==2.0.1
     # via
     #   jinja2
     #   mako
-mlflow==1.22.0
+matplotlib==3.6.3
+    # via mlflow
+mlflow==2.9.1
     # via -r requirements.in
 multidict==5.2.0
     # via
@@ -206,11 +219,14 @@ networkx==2.6.3
 numpy==1.19.5
     # via
     #   -r requirements.in
+    #   contourpy
     #   h5py
     #   keras-preprocessing
+    #   matplotlib
     #   mlflow
     #   opt-einsum
     #   pandas
+    #   pyarrow
     #   scikit-learn
     #   scipy
     #   tensorboard
@@ -222,6 +238,7 @@ opt-einsum==3.3.0
 packaging==21.3
     # via
     #   dvc
+    #   matplotlib
     #   mlflow
     #   pytest
 pandas==1.3.5
@@ -237,17 +254,15 @@ pathtools==0.1.2
 phonenumbers==8.12.40
     # via python-benedict
 pillow==9.0.0
-    # via -r requirements.in
+    # via
+    #   -r requirements.in
+    #   matplotlib
 pluggy==1.0.0
     # via pytest
 ply==3.11
     # via
     #   dvc
     #   jsonpath-ng
-prometheus-client==0.12.0
-    # via prometheus-flask-exporter
-prometheus-flask-exporter==0.18.7
-    # via mlflow
 promise==2.3
     # via wandb
 protobuf==3.19.1
@@ -262,6 +277,8 @@ psutil==5.9.0
     #   wandb
 py==1.11.0
     # via pytest
+pyarrow==14.0.1
+    # via mlflow
 pyasn1==0.4.8
     # via
     #   dvc
@@ -287,6 +304,7 @@ pyparsing==2.4.7
     # via
     #   dvc
     #   grandalf
+    #   matplotlib
     #   packaging
     #   pydot
 pytest==6.2.5
@@ -297,6 +315,7 @@ python-dateutil==2.8.2
     # via
     #   alembic
     #   botocore
+    #   matplotlib
     #   pandas
     #   python-benedict
     #   wandb
@@ -335,18 +354,22 @@ rich==10.16.2
     # via dvc
 rsa==4.8
     # via google-auth
-ruamel.yaml==0.17.20
+ruamel-yaml==0.17.20
     # via dvc
-ruamel.yaml.clib==0.2.6
-    # via ruamel.yaml
+ruamel-yaml-clib==0.2.6
+    # via ruamel-yaml
 s3fs==2021.6.1
     # via dvc
 s3transfer==0.3.7
     # via boto3
 scikit-learn==1.0.2
-    # via -r requirements.in
+    # via
+    #   -r requirements.in
+    #   mlflow
 scipy==1.7.3
-    # via scikit-learn
+    # via
+    #   mlflow
+    #   scikit-learn
 sentry-sdk==1.5.1
     # via wandb
 shortuuid==1.0.8
@@ -399,7 +422,7 @@ tensorboard-plugin-wit==1.8.1
     # via tensorboard
 tensorflow==2.7.0
     # via -r requirements.in
-tensorflow_addons==0.15.0
+tensorflow-addons==0.15.0
     # via -r requirements.in
 tensorflow-estimator==2.7.0
     # via tensorflow
@@ -459,7 +482,7 @@ yarl==1.7.2
     # via aiohttp
 yaspin==2.1.0
     # via wandb
-zc.lockfile==2.0
+zc-lockfile==2.0
     # via dvc
 zipp==3.7.0
     # via importlib-metadata