Skip to content

Commit bd414af

Browse files
committed
fix test
1 parent b6510ac commit bd414af

File tree

1 file changed

+3
-2
lines changed

1 file changed

+3
-2
lines changed

python/raydp/tests/test_data_owner_transfer.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -213,7 +213,7 @@ def test_api_compatibility(ray_cluster, jdk17_extra_spark_configs):
213213
num_executor = 1
214214

215215
spark = raydp.init_spark(
216-
app_name = "example",
216+
app_name = "test_api_compatibility",
217217
num_executors = num_executor,
218218
executor_cores = 1,
219219
executor_memory = "500M",
@@ -227,7 +227,8 @@ def test_api_compatibility(ray_cluster, jdk17_extra_spark_configs):
227227

228228
# check compatibility of ray 1.9.0 API: no data onwership transfer
229229
ds = ray.data.from_spark(df_train)
230-
ds.show(1)
230+
if not ray_client.ray.is_connected():
231+
ds.show(1)
231232
ray_gc() # ensure GC kicked in
232233
time.sleep(3)
233234

0 commit comments

Comments
 (0)