chore: black format

Signed-off-by: Avik Basu <avikbasu93@gmail.com>

.github/workflows/ci.yml

@@ -30,7 +30,7 @@ jobs:
     - name: Install dependencies
       run: |
         poetry env use ${{ matrix.python-version }}
-        poetry install --no-root -E mlflow
+        poetry install --all-extras
 
     - name: Test with pytest
       run: make test

.github/workflows/coverage.yml

@@ -30,7 +30,7 @@ jobs:
     - name: Install dependencies
       run: |
         poetry env use ${{ matrix.python-version }}
-        poetry install --no-root -E mlflow
+        poetry install --all-extras
 
     - name: Run Coverage
       run: |

.github/workflows/lint.yml

@@ -30,7 +30,7 @@ jobs:
     - name: Install dependencies
       run: |
         poetry env use ${{ matrix.python-version }}
-        poetry install --no-root
+        poetry install
 
     - name: Black format check
       run: poetry run black --check .

.github/workflows/pypi.yml

@@ -30,7 +30,7 @@ jobs:
       - name: Install dependencies
         run: |
           poetry env use ${{ matrix.python-version }}
-          poetry install --no-root -E mlflow
+          poetry install --all-extras
 
       - name: Build dist
         run: poetry build

Makefile

@@ -23,7 +23,7 @@ lint: format
 
 # install all dependencies
 setup:
-	poetry install -v
+	poetry install --all-extras
 
 # test your application (tests in the tests/ directory)
 test:

numalogic/tests/registry/_mlflow_utils.py

@@ -110,7 +110,7 @@ def mock_log_model_pytorch(*_, **__):
         signature_dict=None,
         utc_time_created="2022-05-23 22:35:59.557372",
         mlflow_version="1.26.0",
-        signature=None
+        signature=None,
     )
 
 
@@ -134,7 +134,7 @@ def mock_log_model_sklearn(*_, **__):
         signature_dict=None,
         utc_time_created="2022-05-23 22:35:59.557372",
         mlflow_version="1.26.0",
-        signature=None
+        signature=None,
     )