mypy

Lightning-AI · Feb 7, 2023 · 9de3ac7 · 9de3ac7
1 parent d883b7c
commit 9de3ac7
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/src/lightning_utilities/core/enums.py b/src/lightning_utilities/core/enums.py
@@ -69,9 +69,10 @@ def from_str(
                     return cls[enum_key]
 
         warnings.warn(UserWarning(f"Invalid string: expected one of {allowed}, but got {value}."))
+        return None
 
     @classmethod
-    def _allowed_matches(cls, source) -> list:
+    def _allowed_matches(cls, source: str) -> list:
         keys, vals = [], []
         for enum_key, enum_val in cls.__members__.items():
             keys.append(enum_key)