code: def dd_simple():
      dd = pytest.importorskip(
          "dask.dataframe", reason="Dask not installed, skipping"
      )
      es = EntitySet("test")
      df = pd.DataFrame({"foo": [1]})
      df = dd.from_pandas(df, npartitions=2)
      es.add_dataframe(df, dataframe_name="test", index="foo")
      return es
  
file_path: featuretools/featuretools/tests/entityset_tests/test_plotting.py
imports: import os
  import re
  import graphviz
  import pandas
  import pytest
  from featuretools import EntitySet
  from featuretools.utils.gen_utils import Library
label_desc: Use image node_id 2 for calling a global function with example usage: dd_simple() and returns: es