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, )