def _create_tfs_model(self,
role=None,
vpc_config_override=VPC_CONFIG_DEFAULT,
entry_point=None,
source_dir=None,
dependencies=None,
**kwargs):
"""Placeholder docstring"""
# remove image kwarg
if "image" in kwargs:
image = kwargs["image"]
kwargs = {k: v for k, v in kwargs.items() if k != "image"}
else:
image = None
return Model(model_data=self.model_data,
role=role,
image=(image or self.image_name),
name=self._current_job_name,
container_log_level=self.container_log_level,
framework_version=utils.get_short_version(
self.framework_version),
sagemaker_session=self.sagemaker_session,
vpc_config=self.get_vpc_config(vpc_config_override),
entry_point=entry_point,
source_dir=source_dir,
dependencies=dependencies,
enable_network_isolation=self.enable_network_isolation(),
**kwargs)
def _create_tfs_model(self, role=None, vpc_config_override=VPC_CONFIG_DEFAULT):
return Model(
model_data=self.model_data,
role=role,
image=self.image_name,
name=self._current_job_name,
container_log_level=self.container_log_level,
framework_version=utils.get_short_version(self.framework_version),
sagemaker_session=self.sagemaker_session,
vpc_config=self.get_vpc_config(vpc_config_override),
)
def _create_tfs_model(self,
role=None,
vpc_config_override=VPC_CONFIG_DEFAULT,
entry_point=None,
source_dir=None,
dependencies=None,
**kwargs):
"""Placeholder docstring"""
return Model(model_data=self.model_data,
role=role,
container_log_level=self.container_log_level,
framework_version=utils.get_short_version(
self.framework_version),
sagemaker_session=self.sagemaker_session,
vpc_config=self.get_vpc_config(vpc_config_override),
entry_point=entry_point,
source_dir=source_dir,
dependencies=dependencies,
**kwargs)
