fix compile w/o cuda

nbren12 · nbren12 · commit fa2d3ac0108d · 2024-09-16T12:04:16.000-07:00
diff --git a/setup.py b/setup.py
@@ -80,7 +80,7 @@ def get_compiler():
             extra_compile_args={'nvcc': ['-O2']},
         ),
     )
-except ImportError:
+except (ImportError, OSError):
     warnings.warn("Cuda extensions for torch not found, skipping cuda healpix padding module")
 
 

Original file line number	Diff line number	Diff line change
`@@ -80,7 +80,7 @@ def get_compiler():`
`80`	`80`	`extra_compile_args={'nvcc': ['-O2']},`
`81`	`81`	`),`
`82`	`82`	`)`
`83`		`-except ImportError:`
	`83`	`+except (ImportError, OSError):`
`84`	`84`	`warnings.warn("Cuda extensions for torch not found, skipping cuda healpix padding module")`
`85`	`85`
`86`	`86`