From 42361a18016d57b89583bd716b7c0ce79e827bb8 Mon Sep 17 00:00:00 2001
From: SangBin Cho <rkooo567@gmail.com>
Date: Thu, 17 Feb 2022 02:44:36 +0900
Subject: [PATCH] [Test] Fix Dask on Ray 1 TB bug #22431  Open

Fixes a bug. It seems like not df is not working with dataframe
---
 release/nightly_tests/dask_on_ray/dask_on_ray_sort.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/release/nightly_tests/dask_on_ray/dask_on_ray_sort.py b/release/nightly_tests/dask_on_ray/dask_on_ray_sort.py
index 92999e61a..d46e4368b 100644
--- a/release/nightly_tests/dask_on_ray/dask_on_ray_sort.py
+++ b/release/nightly_tests/dask_on_ray/dask_on_ray_sort.py
@@ -57,9 +57,10 @@ def load_dataset(client, data_dir, s3_bucket, nbytes, npartitions):
     df = None
     max_retry = 3
     retry = 0
-    while not df and retry < max_retry:
+    while retry < max_retry:
         try:
             df = dd.read_parquet(filenames)
+            break
         except FileNotFoundError as e:
             print(f"Failed to load a file. {e}")
             # Wait a little bit before retrying.