yujunjun/autogen
1
0
Fork 0
mirror of https://github.com/microsoft/autogen.git synced 2025-10-11 16:08:55 +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

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

@ -16,7 +16,6 @@ import docker
from azure.core.exceptions import ClientAuthenticationError
from azure.identity import DefaultAzureCredential, get_bearer_token_provider
from docker.errors import APIError, DockerException, ImageNotFound
from docker.models.containers import Container
from typing_extensions import TypedDict
from .version import __version__
@ -504,9 +503,7 @@ echo RUN.SH COMPLETE !#!#
assert image is not None
# Create and run the container
container: Container = cast(
Container,
client.containers.run(
container = client.containers.run(
image,
command=["sh", "run.sh"],
working_dir="/workspace",
@ -516,7 +513,6 @@ echo RUN.SH COMPLETE !#!#
auto_remove=True,
# 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.