fix: typing warning

Signed-off-by: Aaron Pham <29749331+aarnphm@users.noreply.github.com>
bentoml · Nov 4, 2022 · 7e23357 · 7e23357
1 parent 3377eb7
commit 7e23357
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/simple_di/providers.py b/simple_di/providers.py
@@ -196,7 +196,7 @@ def set(self, value: Any) -> None:
         for i in self._path[:-1]:
             if isinstance(i, Provider):
                 i = i.get()
-            _next: Union[_SentinelClass, Dict[Any, Any]] = _cursor.get(i, sentinel)
+            _next: Dict[Any, Any] = _cursor.get(i, sentinel)
             if isinstance(_next, _SentinelClass):
                 _next = {}
                 _cursor[i] = _next
@@ -219,6 +219,7 @@ def get(self) -> Any:
             if isinstance(_cursor, dict):
                 _cursor = _cursor.get(i, None)
             else:
+                assert _cursor is not None
                 _cursor = _cursor[i]
         return _cursor