update make_ecommerce_entityset

alteryx · Feb 22, 2024 · adfb08e · adfb08e
1 parent c26f193
commit adfb08e
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/featuretools/tests/testing_utils/mock_ds.py b/featuretools/tests/testing_utils/mock_ds.py
@@ -2,7 +2,6 @@
 
 import numpy as np
 import pandas as pd
-import pytest
 from woodwork.logical_types import (
     URL,
     Boolean,
@@ -24,6 +23,9 @@
 )
 
 from featuretools.entityset import EntitySet
+from featuretools.utils.gen_utils import import_or_none
+
+dask = import_or_none("dask")
 
 
 def make_ecommerce_entityset(with_integer_time_index=False):
@@ -37,8 +39,8 @@ def make_ecommerce_entityset(with_integer_time_index=False):
          \\ /   .
           L     Log
     """
-    dask = pytest.importorskip("dask", reason="Dask not installed, skipping")
-    dask.config.set({"dataframe.convert-string": False})
+    if dask:
+        dask.config.set({"dataframe.convert-string": False})
 
     dataframes = make_ecommerce_dataframes(
         with_integer_time_index=with_integer_time_index,