def get_container(self, settings: SerializationSettings) -> Container:
env = {
**settings.env,
**self.environment
} if self.environment else settings.env
return _get_container_definition(
image=get_registerable_container_image(None,
settings.image_config),
command=[],
args=self.get_command(settings=settings),
data_loading_config=None,
environment=env,
)
def get_container(self, settings: SerializationSettings) -> _task_model.Container:
env = {**settings.env, **self.environment} if self.environment else settings.env
return _get_container_definition(
image=self._image,
command=self._cmd,
args=self._args,
data_loading_config=_task_model.DataLoadingConfig(
input_path=self._input_data_dir,
output_path=self._output_data_dir,
format=self._md_format.value,
enabled=True,
io_strategy=self._io_strategy.value if self._io_strategy else None,
),
environment=env,
)
def get_container(self, settings: SerializationSettings) -> _task_model.Container:
env = {**settings.env, **self.environment} if self.environment else settings.env
return _get_container_definition(
image=get_registerable_container_image(self.container_image, settings.image_config),
command=[],
args=self.get_command(settings=settings),
data_loading_config=None,
environment=env,
storage_request=self.resources.requests.storage,
cpu_request=self.resources.requests.cpu,
gpu_request=self.resources.requests.gpu,
memory_request=self.resources.requests.mem,
storage_limit=self.resources.limits.storage,
cpu_limit=self.resources.limits.cpu,
gpu_limit=self.resources.limits.gpu,
memory_limit=self.resources.limits.mem,
)