| 6 | |
| 7 | |
| 8 | class ResourceRequest: |
| 9 | n_nodes: int = 0 |
| 10 | resources: Dict[str, Union[int, float, str]] = dataclasses.field(default_factory=dict) |
| 11 | min_time: Optional[datetime.timedelta] = None |
| 12 | |
| 13 | def __init__( |
| 14 | self, |
| 15 | *, |
| 16 | n_nodes=0, |
| 17 | cpus: Union[int, float, str] = 1, |
| 18 | resources: Optional[Dict[str, Union[int, float, str]]] = None, |
| 19 | min_time: Optional[datetime.timedelta] = None, |
| 20 | ): |
| 21 | self.n_nodes = n_nodes |
| 22 | if resources is None: |
| 23 | resources = {} |
| 24 | resources["cpus"] = cpus |
| 25 | self.resources = resources |
| 26 | self.min_time = min_time |
| 27 | |
| 28 | def __repr__(self): |
| 29 | return f"<ResourceRequest n_nodes={self.n_nodes} resources={self.resources} min_time={self.min_time}>" |
| 30 | |
| 31 | |
| 32 | @dataclasses.dataclass() |
no outgoing calls