fix sdist to pass preflight check

cupy · Oct 19, 2023 · a63377f · a63377f
1 parent 9a3dec2
commit a63377f
Showing 1 changed file with 9 additions and 2 deletions.
diff --git a/install/cupy_builder/_preflight.py b/install/cupy_builder/_preflight.py
@@ -13,8 +13,15 @@ def preflight_check(ctx: Context) -> bool:
     is_git = os.path.isdir(os.path.join(source_root, '.git'))
     for submodule in ('third_party/cccl',
                       'cupy/_core/include/cupy/jitify'):
-        if 0 < len(os.listdir(os.path.join(source_root, submodule))):
-            continue
+        dirpath = os.path.join(source_root, submodule)
+        if os.path.isdir(dirpath):
+            if 0 < len(os.listdir(dirpath)):
+                continue
+        else:
+            if not is_git:
+                # sdist does not contain third_party directory
+                continue
+
         if is_git:
             msg = f'''
 ===========================================================================