Skip to content

Conversation

ansjindal
Copy link
Contributor

  • Add reservation_config field to LeptonExecutor dataclass
  • Pass reservation_config from executor to job specification
  • Enable users to configure reservation settings for Lepton jobs

- Add reservation_config field to LeptonExecutor dataclass
- Pass reservation_config from executor to job specification
- Enable users to configure reservation settings for Lepton jobs

Signed-off-by: ansjindal <[email protected]>
@ansjindal ansjindal force-pushed the feature/lepton-reservation-config branch from e153be4 to b984812 Compare August 22, 2025 19:02
Copy link
Contributor

@roclark roclark left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Apologies I didn't see this earlier. One note about the type, but this would be great to have this update, thanks for creating it!

lepton_job_dir: str = field(init=False, default="")
custom_spec: dict[str, Any] = field(default_factory=dict)
pre_launch_commands: list[str] = field(default_factory=list) # Custom commands before launch
reservation_config: Optional[dict[str, Any]] = None
Copy link
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Should this be of type str instead of dict? Looking at the Lepton API it appears it just uses string.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

2 participants