[tests] Skip bad ray memory call in shuffle nightly tests if it fails r…

…ay-project#29403

ray memory sometimes fails if the cluster is under load. It's only used in the nightly test for debugging, so skip it if it fails.

release/nightly_tests/dataset/sort.py

@@ -3,6 +3,7 @@
 import resource
 import time
 from typing import List
+import traceback
 
 import numpy as np
 import psutil
@@ -134,7 +135,11 @@ def make_block(count: int, num_columns: int) -> Block:
     rss = int(process.memory_info().rss)
     print(f"rss: {rss / 1e9}/GB")
 
-    print(memory_summary(stats_only=True))
+    try:
+        print(memory_summary(stats_only=True))
+    except Exception:
+        print("Failed to retrieve memory summary")
+        print(traceback.format_exc())
     print("")
 
     print(ds.stats())