Browse Source

[MNT] modify log to add container id info

tags/v0.3.2
bxdd 2 years ago
parent
commit
90bdaa00ee
1 changed files with 5 additions and 3 deletions
  1. +5
    -3
      learnware/client/container.py

+ 5
- 3
learnware/client/container.py View File

@@ -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.")



Loading…
Cancel
Save