diff --git a/integration-tests/conftest.py b/integration-tests/conftest.py index 93ff9e40..b136cbdb 100644 --- a/integration-tests/conftest.py +++ b/integration-tests/conftest.py @@ -517,6 +517,8 @@ def launcher(event_loop): docker.types.DeviceRequest(count=gpu_count, capabilities=[["gpu"]]) ] + print("Starting docker") + container = client.containers.run( DOCKER_IMAGE, command=args, @@ -532,6 +534,10 @@ def launcher(event_loop): shm_size="1G", ) + import time + + time.sleep(600) + yield ContainerLauncherHandle(client, container.name, port) if not use_flash_attention: