FIX-modin-project#2540: add __iter__ implementation

Signed-off-by: Anatoly Myachev <anatoly.myachev@intel.com>
anmyachev · Dec 15, 2020 · 47a95e3 · 47a95e3
1 parent da83b62
commit 47a95e3
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 1 deletion.
diff --git a/modin/config/envvars.py b/modin/config/envvars.py
@@ -233,7 +233,7 @@ def _check_vars():
         and issubclass(obj, EnvironmentVariable)
         and not obj.is_abstract
     }
-    found_names = {name for name in os.environ.keys() if name.startswith("MODIN_")}
+    found_names = {name for name in os.environ if name.startswith("MODIN_")}
     unknown = found_names - valid_names
     if unknown:
         warnings.warn(

diff --git a/modin/conftest.py b/modin/conftest.py
@@ -154,6 +154,9 @@ def __contains__(self, name):
         def __getattr__(self, name):
             return getattr(orig_env, name)
 
+        def __iter__(self):
+            return iter(orig_env)
+
     os.environ = PatchedEnv()
     yield
     os.environ = orig_env