diff --git a/.github/workflows/build.yaml b/.github/workflows/build.yaml index 99ec299e..81707088 100644 --- a/.github/workflows/build.yaml +++ b/.github/workflows/build.yaml @@ -199,5 +199,5 @@ jobs: echo $DOCKER_IMAGE # python -c 'import docker; client = docker.from_env(); client.containers.run("registry.internal.huggingface.tech/api-inference/community/text-generation-inference:sha-318bbc7-rocm", devices=["/dev/kfd", "/dev/dri"], env={"MODEL_ID": "gpt2"}, detach=True)' which pytest - pytest -s -vv integration-tests -k flash_llama_grammar_tools + pytest -s -vv integration-tests -k flash_llama_grammar_tools_auto # pytest -s -vv integration-tests ${PYTEST_FLAGS} ${EXTRA_PYTEST} diff --git a/integration-tests/conftest.py b/integration-tests/conftest.py index 7814722b..2808b950 100644 --- a/integration-tests/conftest.py +++ b/integration-tests/conftest.py @@ -532,7 +532,7 @@ def launcher(event_loop): container = client.containers.run( DOCKER_IMAGE, command=args, - name=container_name, + # name=container_name, environment=env, auto_remove=False, detach=True,