File tree Expand file tree Collapse file tree 1 file changed +9
-0
lines changed
Expand file tree Collapse file tree 1 file changed +9
-0
lines changed Original file line number Diff line number Diff line change 66
77import ray
88import ray .data
9+ from ray .data import DataContext
910
1011from graphgen .bases import Config , Node
1112from graphgen .utils import logger
@@ -20,6 +21,14 @@ def __init__(
2021 self .functions = functions
2122 self .datasets : Dict [str , ray .data .Dataset ] = {}
2223
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+
2332 if not ray .is_initialized ():
2433 context = ray .init (
2534 ignore_reinit_error = True ,
You can’t perform that action at this time.
0 commit comments