diff --git a/graphgen/engine.py b/graphgen/engine.py
index 62ab5281..501aa854 100644
--- a/graphgen/engine.py
+++ b/graphgen/engine.py
@@ -6,6 +6,7 @@
 
 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 @@ def __init__(
         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,