Fix linting & support for python 3.8

SeldonIO · Jul 12, 2023 · 552e754 · 552e754
1 parent 189bf7c
commit 552e754
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 5 deletions.
diff --git a/mlserver/sys_tracing/tracepoints.py b/mlserver/sys_tracing/tracepoints.py
@@ -21,7 +21,7 @@
 See `system_tracing.provider` for more details about the actual tracepoint
 implementation.
 """
-from typing import Optional
+from typing import Optional, Tuple, List
 from enum import Enum, IntEnum, auto
 
 # The underlying SDT tracepoint implementation currently only works on x86-64
@@ -120,7 +120,7 @@ def all():
     def none():
         return set()
 
-    def get_arg_types(self) -> tuple[Optional[list[int]], ArgStatus]:
+    def get_arg_types(self) -> Tuple[Optional[List[int]], ArgStatus]:
         """
         Returns a (arg_types_list, status) tuple for the current tracepoint. The
         arg_types_list, when not None, defines the function prototype of the

diff --git a/mlserver/types/base.py b/mlserver/types/base.py
@@ -1,5 +1,4 @@
-from pydantic import BaseModel as _BaseModel, json as pydantic_json
-from enum import Enum
+from pydantic import BaseModel as _BaseModel
 
 
 class BaseModel(_BaseModel):

diff --git a/tests/test_settings.py b/tests/test_settings.py
@@ -7,7 +7,6 @@
 from mlserver.repository import DEFAULT_MODEL_SETTINGS_FILENAME
 
 from .conftest import TESTDATA_PATH, TESTS_PATH
-from mlserver.types import Tracepoint
 
 
 def test_settings_from_env(monkeypatch):