fixes, linting and such

Author: omesser

mlem/utils/module.py

@@ -8,9 +8,9 @@
 import sys
 import threading
 import warnings
-from importlib.metadata import distribution, PackageNotFoundError
 from collections import defaultdict
 from functools import lru_cache, wraps
+from importlib.metadata import PackageNotFoundError, distribution
 from pickle import PickleError
 from types import FunctionType, LambdaType, MethodType, ModuleType
 from typing import Dict, List, Optional, Set, Union

setup.py

@@ -59,7 +59,7 @@
     "nbloader",
     # We're using regex to test requirement version extraction
     # edge case, see: https://github.com/iterative/mlem/issues/688
-    'regex==2023.6.3',
+    "regex==2023.6.3",
 ]
 
 extras = {
@@ -78,7 +78,7 @@
     "lightgbm": ["lightgbm"],
     "fastapi": ["uvicorn", "fastapi"],
     "prometheus": ["prometheus-fastapi-instrumentator"],
-    "streamlit": ["uvicorn", "fastapi", "streamlit>=1.14.0", "streamlit_pydantic"],
+    "streamlit": ["uvicorn", "fastapi", "streamlit", "streamlit_pydantic"],
     "sagemaker": ["docker", "boto3", "sagemaker"],
     "torch": ["torch"],
     "tensorflow": ["tensorflow"],

tests/utils/test_module_tools.py

@@ -1,11 +1,11 @@
 import os
 import subprocess
 from typing import ClassVar
-import regex
 
 import nbformat
 import numpy
 import pytest
+import regex
 from pydantic import BaseModel
 
 from mlem.core.requirements import Requirements

tests/utils/test_save.ipynb

@@ -78,4 +78,4 @@
  },
  "nbformat": 4,
  "nbformat_minor": 0
-}
+}