Skip to content

Commit 778239c

Browse files
committed
chore: fix imports on windows
1 parent 6583610 commit 778239c

File tree

1 file changed

+3
-2
lines changed

1 file changed

+3
-2
lines changed

cuda_bindings/tests/test_cufile.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -14,7 +14,6 @@
1414

1515
import cuda.bindings.driver as cuda
1616
import pytest
17-
from cuda.bindings.cufile import cuFileError
1817

1918
# Configure logging to show INFO level and above
2019
logging.basicConfig(
@@ -26,7 +25,9 @@
2625
try:
2726
from cuda.bindings import cufile
2827
except ImportError:
29-
cufile = None
28+
cufile = cuFileError = None
29+
else:
30+
from cuda.bindings.cufile import cuFileError
3031

3132

3233
def platform_is_wsl():

0 commit comments

Comments
 (0)