def __init__(
self,
base_scheduler: Optional[TrialScheduler] = None,
resources_allocation_function: Optional[
Callable[
[
"trial_runner.TrialRunner",
Trial,
Dict[str, Any],
"ResourceChangingScheduler",
],
Union[None, PlacementGroupFactory, Resources],
]
] = _DistributeResourcesDefault,
) -> None:
super().__init__()
if resources_allocation_function is None:
warnings.warn(
"`resources_allocation_function` is None. No resource "
"requirements will be changed at any time. Pass a "
"correctly defined function to enable functionality."
)
self._resources_allocation_function = resources_allocation_function
self._base_scheduler = base_scheduler or FIFOScheduler()
self._base_trial_resources: Optional[
Union[Resources, PlacementGroupFactory]
] = None
self._trials_to_reallocate: Dict[
Trial, Union[None, dict, PlacementGroupFactory]
] = {}
self._reallocated_trial_ids: Set[str] = set()
self._metric = None
self._mode = None
def trial_scheduler(self):
return FIFOScheduler()