From 5806deab1560343cafa7dd26ed188dea33cda8e5 Mon Sep 17 00:00:00 2001 From: Clayton Parnell Date: Fri, 16 Feb 2024 16:43:52 -0800 Subject: [PATCH] fix: Fix autogluon tests for codebuild --- test/test_artifacts/v1/scripts/run_autogluon_tests.sh | 2 +- test/test_dockerfile_based_harness.py | 3 ++- 2 files changed, 3 insertions(+), 2 deletions(-) diff --git a/test/test_artifacts/v1/scripts/run_autogluon_tests.sh b/test/test_artifacts/v1/scripts/run_autogluon_tests.sh index 036d1432..9a669143 100644 --- a/test/test_artifacts/v1/scripts/run_autogluon_tests.sh +++ b/test/test_artifacts/v1/scripts/run_autogluon_tests.sh @@ -1,6 +1,6 @@ #!/bin/bash -AUTOGLUON_VERSION=$(micromamba list | grep autogluon | tr -s ' ' | cut -d ' ' -f 3) +AUTOGLUON_VERSION=$(micromamba list | grep autogluon | tr -s ' ' | cut -d ' ' -f 3 | head -n 1) git checkout tags/v$AUTOGLUON_VERSION # Run autogluon quick start as end-to-end check diff --git a/test/test_dockerfile_based_harness.py b/test/test_dockerfile_based_harness.py index bbaa4575..b3d910f3 100644 --- a/test/test_dockerfile_based_harness.py +++ b/test/test_dockerfile_based_harness.py @@ -126,7 +126,8 @@ def _validate_docker_images(dockerfile_path: str, required_packages: List[str], image, _ = _docker_client.images.build(path=test_artifacts_path, dockerfile=dockerfile_path, tag=dockerfile_path.lower().replace('.', '-'), - rm=True, buildargs={'COSMOS_IMAGE': docker_image_identifier}) + rm=True, buildargs={'COSMOS_IMAGE': docker_image_identifier}, + shm_size='256m') except BuildError as e: for line in e.build_log: if 'stream' in line: