Auto-sync from demo at Fri Dec 19 08:38:09 UTC 2025

graphgen/engine.py

@@ -6,6 +6,7 @@ from typing import Any, Callable, Dict, List, Set
 
 import ray
 import ray.data
+from ray.data import DataContext
 
 from graphgen.bases import Config, Node
 from graphgen.utils import logger
@@ -20,6 +21,14 @@ class Engine:
         self.functions = functions
         self.datasets: Dict[str, ray.data.Dataset] = {}
 
+        ctx = DataContext.get_current()
+        ctx.enable_rich_progress_bars = False
+        ctx.use_ray_tqdm = False
+        # Disable tensor extension casting to avoid conversion errors with complex types
+        # (e.g., gene_synonyms, gene_names which are lists/arrays)
+        ctx.enable_tensor_extension_casting = False
+        ctx._metrics_export_port = 0  # Disable metrics exporter to avoid RpcError
+
         if not ray.is_initialized():
             context = ray.init(
                 ignore_reinit_error=True,