diff --git a/learnware/client/container.py b/learnware/client/container.py index 8d9caae..6c9bdf1 100644 --- a/learnware/client/container.py +++ b/learnware/client/container.py @@ -86,6 +86,7 @@ class ModelCondaContainer(ModelContainer): def _init_env(self): install_environment(self.learnware_dirpath, self.conda_env) + logger.info(f"Conda env {self.conda_env} is generated.") def _remove_env(self): remove_enviroment(self.conda_env) @@ -222,6 +223,7 @@ class ModelDockerContainer(ModelContainer): "pids_limit": -1, } container = client.containers.run(**container_config) + logger.info(f"Docker container {container.id[:12]} is generated.") try: environment_cmd = [ @@ -263,13 +265,13 @@ class ModelDockerContainer(ModelContainer): if isinstance(docker_container, docker.models.containers.Container): client = docker.from_env() container_ids = [container.id for container in client.containers.list()] - + if docker_container.id in container_ids: docker_container.stop() docker_container.remove() - logger.info(f"Docker container {docker_container.id} is stopped and removed.") + logger.info(f"Docker container {docker_container.id[:12]} is stopped and removed.") else: - logger.info(f"Docker container {docker_container.id} has already been removed.") + logger.info(f"Docker container {docker_container.id[:12]} has already been removed.") else: logger.error("Type of docker_container is not docker.models.containers.Container.")