Run formatter

SeldonIO · Apr 8, 2021 · 2a02666 · 2a02666
1 parent 02fdc71
commit 2a02666
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 15 deletions.
diff --git a/tempo/mlserver.py b/tempo/mlserver.py
@@ -4,8 +4,8 @@
 
 from .serve.base import BaseModel
 from .serve.loader import load
-from .serve.utils import PredictMethodAttr
 from .serve.metadata import ModelFramework
+from .serve.utils import PredictMethodAttr
 
 
 def _is_class(model: BaseModel) -> bool:

diff --git a/tempo/serve/loader.py b/tempo/serve/loader.py
@@ -11,11 +11,7 @@
 import yaml
 
 from tempo.conf import settings
-from tempo.serve.constants import (
-    DefaultModelFilename,
-    DefaultRemoteFilename,
-    MLServerEnvDeps,
-)
+from tempo.serve.constants import DefaultModelFilename, DefaultRemoteFilename, MLServerEnvDeps
 from tempo.utils import logger
 
 
@@ -61,9 +57,7 @@ def _get_env(conda_env_file_path: str = None, env_name: str = None) -> dict:
     return env
 
 
-def save_environment(
-    conda_pack_file_path: str, conda_env_file_path: str = None, env_name: str = None
-):
+def save_environment(conda_pack_file_path: str, conda_env_file_path: str = None, env_name: str = None):
     if env_name:
         _pack_environment(env_name, conda_pack_file_path)
     else:
@@ -93,9 +87,7 @@ def _has_required_deps(env: dict) -> bool:
     for dep in MLServerEnvDeps:
         parts = re.split(r"==|>=|<=|~=|!=|>|<|==:", dep)
         module = parts[0]
-        r = re.compile(
-            fr"{module}$|({module}((==|>=|<=|~=|!=|>|<|==:)[0-9]+\.[0-9]+.[0-9]+))"
-        )
+        r = re.compile(fr"{module}$|({module}((==|>=|<=|~=|!=|>|<|==:)[0-9]+\.[0-9]+.[0-9]+))")
         newlist = list(filter(r.match, pip_deps["pip"]))
         if len(newlist) == 0:
             return False
@@ -116,9 +108,7 @@ def _add_required_deps(env: dict) -> dict:
     for dep in MLServerEnvDeps:
         parts = re.split(r"==|>=|<=|~=|!=|>|<|==:", dep)
         module = parts[0]
-        r = re.compile(
-            fr"{module}$|({module}((==|>=|<=|~=|!=|>|<|==:)[0-9]+\.[0-9]+.[0-9]+))"
-        )
+        r = re.compile(fr"{module}$|({module}((==|>=|<=|~=|!=|>|<|==:)[0-9]+\.[0-9]+.[0-9]+))")
         newlist = list(filter(r.match, pip_deps["pip"]))
         if len(newlist) == 0:
             pip_deps["pip"].extend(MLServerEnvDeps)