yujunjun/autogen
1
0
Fork 0
mirror of https://github.com/microsoft/autogen.git synced 2025-10-12 00:20:50 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fix lint issue in agbench (#409)

Browse Source 
* Fix lint issue in agbench

* remove import
This commit is contained in:
Jack Gerrits 2024-08-27 14:09:30 -04:00 committed by GitHub
parent 62e5e5b88d
commit 4eadecf4c9
1 changed files with 10 additions and 14 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

24
python/tools/agbench/src/agbench/run_cmd.py
View File

@ -16,7 +16,6 @@ import docker
from azure.core.exceptions import ClientAuthenticationError from azure.core.exceptions import ClientAuthenticationError
from azure.identity import DefaultAzureCredential, get_bearer_token_provider from azure.identity import DefaultAzureCredential, get_bearer_token_provider
from docker.errors import APIError, DockerException, ImageNotFound from docker.errors import APIError, DockerException, ImageNotFound
from docker.models.containers import Container
from typing_extensions import TypedDict from typing_extensions import TypedDict
from .version import __version__ from .version import __version__
@ -504,19 +503,16 @@ echo RUN.SH COMPLETE !#!#
assert image is not None assert image is not None
# Create and run the container # Create and run the container
container: Container = cast( container = client.containers.run(
Container, image,
client.containers.run( command=["sh", "run.sh"],
image, working_dir="/workspace",
command=["sh", "run.sh"], environment=dict(env),
working_dir="/workspace", detach=True,
environment=dict(env), remove=True,
detach=True, auto_remove=True,
remove=True, # Type hint of docker is wrong here
auto_remove=True, volumes=volumes, # type: ignore
# Type hint of docker is wrong here
volumes=volumes, # type: ignore
),
) )
# Read the logs in a streaming fashion. Keep an eye on the time to make sure we don't need to stop. # Read the logs in a streaming fashion. Keep an eye on the time to make sure we don't need to stop.