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)