Fixing another issue with imports

Signed-off-by: Julio Faracco <[email protected]>
discovery-unicamp · Jun 20, 2023 · 57149e6 · 57149e6
1 parent ad02064
commit 57149e6
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/dasf/pipeline/executors/dask.py b/dasf/pipeline/executors/dask.py
@@ -1,9 +1,9 @@
 #!/usr/bin/env python3
 
 import os
-import rmm
 
 try:
+    import rmm
     import cupy as cp
     GPU_SUPPORTED = isinstance(cp.__version__, str)
 except ImportError: