From 12e1b37b093b860b68945a0ed3c8db6ca8bc760f Mon Sep 17 00:00:00 2001 From: Krishan Bhasin Date: Wed, 1 Nov 2023 15:36:17 +0000 Subject: [PATCH] Improve typing on `Environment` class (#1469) Added return type to `__enter__()` --- mlserver/env.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/mlserver/env.py b/mlserver/env.py index 56f7c6c7a..3ddf91537 100644 --- 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"]