Merge pull request #11 from nederlandsespoorwegen/bump/versions

Allow python versions higher than 3.10

.github/workflows/build.yml

@@ -11,12 +11,12 @@ jobs:
     runs-on: ubuntu-latest
     steps:
       - uses: actions/checkout@v2
-      - uses: actions/setup-python@v2
+      - uses: actions/setup-python@v5
         with:
-          python-version: 3.9
-      - uses: Gr1N/setup-poetry@v8
+          python-version: "3.10"
+      - uses: Gr1N/setup-poetry@v9
         with:
-          poetry-version: "1.3.1"
+          poetry-version: "1.7.1"
       - run: poetry install
       - run: poetry build --format wheel
       - uses: ncipollo/release-action@v1

.github/workflows/lint_and_test.yml

@@ -14,12 +14,12 @@ jobs:
     runs-on: ubuntu-latest
     steps:
       - uses: actions/checkout@v2
-      - uses: actions/setup-python@v2
+      - uses: actions/setup-python@v5
         with:
-          python-version: 3.9
-      - uses: Gr1N/setup-poetry@v8
+          python-version: "3.10"
+      - uses: Gr1N/setup-poetry@v9
         with:
-          poetry-version: "1.3.1" 
+          poetry-version: "1.7.1" 
       - run: poetry install
       - run: poetry run black --check .
       - run: poetry run pylint --fail-under=9 */

environment_mlflow_client/env_mlflow_client.py

@@ -1,4 +1,5 @@
 """mlflow client that is aware of the application environment. The main target is Databricks mlflow within one workspace."""
+
 import os
 from collections import defaultdict
 from typing import Any, Dict, List, Optional, Tuple
@@ -29,8 +30,9 @@ def __init__(
     ):
         """
         Create an EnvMlflowClient instance scoped for one logical environment.
-        The environment is set by the environment variable MLFLOW_ENV or by passing the env_name argument.
-        note: On Databricks we do not provide the arguments and rely on environment variables.
+        The environment is set by the environment variable MLFLOW_ENV or by
+        passing the env_name argument. note: On Databricks we do not provide the
+        arguments and rely on environment variables.
 
         Args:
             env_name: environment name overrides the environment variable MLFLOW_ENV
@@ -131,7 +133,7 @@ def create_model_version(
             tags: Tags to set on the model
             run_link: Link to the run that created the model
             description: Description of the model
-            await_creation_for: Number of seconds to wait for the model version to finish being created.
+            await_creation_for: # seconds to wait for the model version to finish being created.
         Returns:
             mlflow ModelVersion
         """
@@ -253,7 +255,7 @@ def log_model_helper(
             kwargs: parameters for model_flavor.log_model, see documentation of model_flavor
             i.e. mlflow.sklearn.log_model
         Returns:
-            Tuple of a mlflow.entities.model_registry.ModelVersion object and mlflow.models.model.ModelInfo object.
+            Mlflow ModelVersion and ModelInfo
 
         """
         if "artifact_path" in kwargs:

pyproject.toml

@@ -9,16 +9,16 @@ packages = [{include = "environment_mlflow_client"}]
 homepage = "https://github.com/nederlandsespoorwegen/environment_mlflow_client"
 
 [tool.poetry.dependencies]
-python = "^3.9,<3.11"
+python = "^3.10"
 mlflow = "^2.1.0"
 
 [tool.poetry.group.dev.dependencies]
 psutil = "^5.9.4"
 pytest = "^7.2.0"
 black = "^23.1.0"
 pylint = "^2.15.9"
-llvmlite = "==0.38.0"  # Databricks runtime 12.1
-numba = "==0.55.1"  # Databricks runtime 12.1
+llvmlite = "==0.39.1"  # Databricks runtime 14.3
+numba = "==0.56.4"  # Databricks runtime 14.3
 coverage = "*"
 pytest-cov = "*"
 

tests/conftest.py

@@ -1,6 +1,7 @@
 """
 session wide MLFlow instance fixture
 """
+
 import os
 import subprocess
 import tempfile

tests/test_custom_model_with_artifacts.py

@@ -19,7 +19,7 @@
 class FakeModel(mlflow.pyfunc.PythonModel):
     """Fake model to log"""
 
-    def predict(self, context, model_input):
+    def predict(self, context, model_input):  # pylint: disable=unused-argument
         return None
 
 

tests/test_env_mlflow_client.py

@@ -10,7 +10,7 @@
 class FakeModel(mlflow.pyfunc.PythonModel):
     """Fake model to log"""
 
-    def predict(self, context, model_input):
+    def predict(self, context, model_input):  # pylint: disable=unused-argument
         return None