Use HAS_GPU in dispatch module to avoid some imports

Author: oliverholworthy

merlin/core/dispatch.py

@@ -26,33 +26,30 @@
 
 from merlin.core.compat import HAS_GPU
 
-try:
-    import cudf
-    import cupy as cp
-    import dask_cudf
-    import rmm
-    from cudf.core.column import as_column, build_column
+cp = None
+cudf = None
+rmm = None
 
+if HAS_GPU:
     try:
-        # cudf >= 21.08
-        from cudf.api.types import is_list_dtype as cudf_is_list_dtype
-        from cudf.api.types import is_string_dtype as cudf_is_string_dtype
+        import cudf
+        import cupy as cp
+        import dask_cudf
+        import rmm
+        from cudf.core.column import as_column, build_column
+
+        try:
+            # cudf >= 21.08
+            from cudf.api.types import is_list_dtype as cudf_is_list_dtype
+            from cudf.api.types import is_string_dtype as cudf_is_string_dtype
+        except ImportError:
+            # cudf < 21.08
+            from cudf.utils.dtypes import is_list_dtype as cudf_is_list_dtype
+            from cudf.utils.dtypes import is_string_dtype as cudf_is_string_dtype
+
     except ImportError:
-        # cudf < 21.08
-        from cudf.utils.dtypes import is_list_dtype as cudf_is_list_dtype
-        from cudf.utils.dtypes import is_string_dtype as cudf_is_string_dtype
+        pass
 
-except ImportError:
-    cp = None
-    cudf = None
-    rmm = None
-
-try:
-    # Dask >= 2021.5.1
-    from dask.dataframe.core import hash_object_dispatch
-except ImportError:
-    # Dask < 2021.5.1
-    from dask.dataframe.utils import hash_object_dispatch
 
 try:
     import nvtx
@@ -71,6 +68,14 @@ def inner2(*args, **kwargs):
         return inner1
 
 
+try:
+    # Dask >= 2021.5.1
+    from dask.dataframe.core import hash_object_dispatch
+except ImportError:
+    # Dask < 2021.5.1
+    from dask.dataframe.utils import hash_object_dispatch
+
+
 if HAS_GPU:
     DataFrameType = Union[pd.DataFrame, cudf.DataFrame]  # type: ignore
     SeriesType = Union[pd.Series, cudf.Series]  # type: ignore