Debug

torchmdnet/neighbors/__init__.py

@@ -2,7 +2,7 @@
 import torch as pt
 from torch.utils import cpp_extension
 
-sources = ['neighbors.cpp', 'neighbors_cpu.cpp'] + ['neighbors_cuda.cu'] if pt.cuda.is_available() else []
+sources = ['neighbors.cpp', 'neighbors_cpu.cpp'] + (['neighbors_cuda.cu'] if pt.cuda.is_available() else [])
 sources = [os.path.join(os.path.dirname(__file__), name) for name in sources]
 print(__file__)
 print(os.path.dirname(__file__))