Land 'to_std' utility parser #93209

fduwjj · fduwjj · commit 68c93ddfa76c · 2023-02-03T02:41:33.000Z
[ghstack-poisoned]
diff --git a/torch/distributed/elastic/multiprocessing/api.py b/torch/distributed/elastic/multiprocessing/api.py
@@ -117,11 +117,10 @@ def from_str(cls, vm: str) -> Union["Std", Dict[int, "Std"]]:
         Any other input raises an exception
         """
 
-        def to_std(v):
-            v = int(v)
+        def to_std(v: str) -> Std: # type: ignore[return]
+            s = Std(int(v))
             for s in Std:
-                if s == v:
-                    return s
+                return s
             # return None -> should NEVER reach here since we regex check input
 
         if re.match(_VALUE_REGEX, vm):  # vm is a number (e.g. 0)
