Skip to content
Open
Changes from 1 commit
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
6 changes: 4 additions & 2 deletions data_juicer/core/executor/ray_executor_partitioned.py
Original file line number Diff line number Diff line change
Expand Up @@ -155,7 +155,7 @@ def __init__(self, cfg: Optional[Namespace] = None):
super().__init__(cfg)

self.executor_type = "ray_partitioned"
self.work_dir = self.cfg.work_dir
self.work_dir = os.path.abspath(self.cfg.work_dir)
Comment thread
cmgzn marked this conversation as resolved.
Outdated
self.job_id = self.cfg.get("job_id", None)

# Initialize temporary directory for Ray operations
Expand All @@ -177,7 +177,9 @@ def __init__(self, cfg: Optional[Namespace] = None):

# Checkpoint configuration and manager initialization
checkpoint_cfg = getattr(self.cfg, "checkpoint", None)
checkpoint_dir = getattr(self.cfg, "checkpoint_dir", os.path.join(self.work_dir, "checkpoints"))
checkpoint_dir = os.path.abspath(
getattr(self.cfg, "checkpoint_dir", os.path.join(self.work_dir, "checkpoints"))
)
Comment thread
cmgzn marked this conversation as resolved.
Outdated

if checkpoint_cfg:
# Use ConfigAccessor to handle both dict and object configurations
Expand Down
Loading