We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 658397d commit c9b065cCopy full SHA for c9b065c
1 file changed
graphgen/engine.py
@@ -6,6 +6,7 @@
6
7
import ray
8
import ray.data
9
+from ray.data import DataContext
10
11
from graphgen.bases import Config, Node
12
from graphgen.utils import logger
@@ -20,6 +21,14 @@ def __init__(
20
21
self.functions = functions
22
self.datasets: Dict[str, ray.data.Dataset] = {}
23
24
+ ctx = DataContext.get_current()
25
+ ctx.enable_rich_progress_bars = False
26
+ ctx.use_ray_tqdm = False
27
+ # Disable tensor extension casting to avoid conversion errors with complex types
28
+ # (e.g., gene_synonyms, gene_names which are lists/arrays)
29
+ ctx.enable_tensor_extension_casting = False
30
+ ctx._metrics_export_port = 0 # Disable metrics exporter to avoid RpcError
31
+
32
if not ray.is_initialized():
33
context = ray.init(
34
ignore_reinit_error=True,
0 commit comments