fix: Fix autogluon tests for codebuild

aws · Feb 17, 2024 · 5c25bb0 · 5c25bb0
1 parent 0ea7f62
commit 5c25bb0
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/src/config.py b/src/config.py
@@ -9,7 +9,7 @@
         'additional_packages_env_in_file': 'gpu.additional_packages_env.in',
         'image_tag_generator': '{image_version}-gpu',
         'env_out_filename': 'gpu.env.out',
-        'pytest_flags': ['--use-gpu'],
+        'pytest_flags': ['--use-gpu', '--shm-size', '256m'],
         'image_type': 'gpu'
     },
     {

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