alteryx / alteryx/evalml

DaskEngine: Repeated Use with Context Manager

Offen
#2,726 1 Kommentar 0 Reaktionen 0 zugewiesene Personen Auf GitHub ansehen
enhancement
Vorherrschende Sprache
Python
Sterne
850
Forks
96
PR-Merge-Kennzahlen
Keine gemergten PRs in 30 T.

Beschreibung

Per #2667 , it was noted that a DaskEngine can't be used in a context manager as the manager will shutdown the associated cluster. We should probably modify the `__enter__()` function to restart the cluster if it's shutdown.

```
def test_automl_dask_called_twice(
X_y_binary_cls,
):
"""Making sure that the max_iterations parameter limits the number of pipelines run."""

X, y = X_y_binary_cls
max_iterations = 4
engine = DaskEngine()
with engine:
par_automl = AutoMLSearch(
X_train=X,
y_train=y,
problem_type="binary",
engine=engine,
max_iterations=max_iterations,
)
par_automl.search()

with engine:
par_automl = AutoMLSearch(
X_train=X,
y_train=y,
problem_type="binary",
engine=engine,
max_iterations=max_iterations,
)
par_automl.search()
```

The above code snippet should be modified into a test that passes.

Beitragsleitfaden

Beitragsleitfaden öffnen

Bewertung

Dieses Issue wurde noch nicht bewertet.

Neue Issues direkt in Ihr Postfach

Eine kurze Übersicht über anfängerfreundliche GitHub-Issues.