fix iter

airgproducts · Nov 26, 2023 · cc49d11 · cc49d11
1 parent c779085
commit cc49d11
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/stubs.py b/stubs.py
@@ -8,8 +8,10 @@
 
 for dir in sys.path:
     print(dir)
-    for subdir in os.listdir(dir):
-        print(f"    {subdir}")
+    basepath = pathlib.Path(dir)
+    if basepath.is_dir():
+        for subdir in basepath.iterdir():
+            print(f"    {subdir}")
 
 import mypy.stubgen