Improve typing on Environment class (#1469)

Added return type to `__enter__()`
SeldonIO · Nov 1, 2023 · 12e1b37 · 12e1b37
1 parent 554fe37
commit 12e1b37
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/mlserver/env.py b/mlserver/env.py
@@ -114,7 +114,7 @@ def _lib_path(self) -> str:
 
         return ""
 
-    def __enter__(self):
+    def __enter__(self) -> "Environment":
         self._prev_sys_path = sys.path
         self._prev_bin_path = os.environ["PATH"]