def create_or_get_image(self): image_params = self.__config['image'] images = ContainerImage.list(self.__ws, image_name=image_config['name']) image = images.find_by_property('version', image_config['version']) if image: return image image_config = ContainerImage.image_configuration( execution_script="score.py", runtime="python", conda_file="config_deploy/myenv.yml", docker_file="config_deploy/Dockerfile", enable_gpu=True, dependencies=[ 'generation', 'config.py', 'skipthoughts_vectors', 'generate.py', 'preprocessing/text_moderator.py' ]) image = ContainerImage.create(name=image_params['name'], models=[], image_config=image_config, workspace=self.__ws) image.wait_for_creation(show_output=True) return image
def getContainerImage(workspace, imageName): image = None images = ContainerImage.list(workspace) for img in images: if img.name == imageName: image = img return image