Cannot import pscan_cuda #19

In 2DMambaMIL/models/pscan_2d.py, the import of pscan_cuda fails. The code falls back to the exception:
try:
from models import pscan_cuda
except:
print('Cannot import pscan_cuda')
There is nothing in that dir
Do you know how to solve this problem?