Skip to content

Commit 5bf6427

Browse files
committed
Updates test to prevent hangs
1 parent 47f4ed0 commit 5bf6427

File tree

1 file changed

+9
-0
lines changed

1 file changed

+9
-0
lines changed

tests/test_evaluation.py

Lines changed: 9 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -13,6 +13,9 @@
1313
import pytest
1414
from datasetsforecast.evaluation import accuracy as ds_evaluate
1515
from pyspark.sql import SparkSession
16+
from dask.distributed import Client
17+
from fugue_dask import DaskExecutionEngine
18+
import dask
1619

1720
import utilsforecast.processing as ufp
1821
from utilsforecast.data import generate_series
@@ -280,6 +283,12 @@ def test_distributed_evaluate(setup_series):
280283
level = [80, 95]
281284
spark = SparkSession.builder.getOrCreate()
282285
spark.sparkContext.setLogLevel("FATAL")
286+
287+
# Use processes=False to avoid deadlocks in CI
288+
client = Client(processes=False)
289+
engine = DaskExecutionEngine(client)
290+
291+
dask.config.set({"dataframe.shuffle.method": "tasks", "scheduler": "synchronous"})
283292
dask_df = dd.from_pandas(setup_series, npartitions=2)
284293
spark_df = spark.createDataFrame(setup_series).repartition(2)
285294
for distributed_df, use_train in product([dask_df, spark_df], [True, False]):

0 commit comments

Comments
 (0)