Set HAS_GPU = False in dispatch if relevant pkgs fail to import (#…

…112) Have tested this fix locally and it works. Now it is possible to import nvtabular without GPU libraries installed
NVIDIA-Merlin · Aug 3, 2022 · ad142af · ad142af
1 parent 1354dcf
commit ad142af
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/merlin/core/dispatch.py b/merlin/core/dispatch.py
@@ -48,7 +48,7 @@
             from cudf.utils.dtypes import is_string_dtype as cudf_is_string_dtype
 
     except ImportError:
-        pass
+        HAS_GPU = False
 
 
 try: