95a1f56a4c0451acd93560af104e642d53db53e8,featuretools/tests/conftest.py,,dask_es,#Any#,53

Before Change

@pytest.fixture
def dask_es(make_es):
    dask_es = copy.deepcopy(make_es)
    for entity in dask_es.entities:
        entity.df = dd.from_pandas(entity.df.reset_index(drop=True), npartitions=2)
    return dask_es


@pytest.fixture

After Change

@pytest.fixture
def dask_es(make_es):
    es = ft.EntitySet(id=make_es.id)
    for entity in make_es.entities:
        es.entity_from_dataframe(entity.id,
                                 dd.from_pandas(entity.df.reset_index(drop=True), npartitions=4),
                                 index=entity.index,
                                 time_index=entity.time_index,
                                 variable_types=entity.variable_types,
                                 secondary_time_index=entity.secondary_time_index)

    for rel in make_es.relationships:
        es.add_relationship(ft.Relationship(es[rel.parent_entity.id][rel.parent_variable.id],
                                            es[rel.child_entity.id][rel.child_variable.id]))
    return es

Instances