fix: Fix autogluon tests for codebuild

aws · Feb 17, 2024 · 5806dea · 5806dea
1 parent 0ea7f62
commit 5806dea
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git 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
@@ -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: