Added comment in autogluon test script. (#525)

aws · Dec 4, 2024 · 6278924 · 6278924
2 parents 52b7570 + 3e3ecb2
commit 6278924
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 0 deletions.
diff --git a/test/test_artifacts/v0/scripts/run_autogluon_tests.sh b/test/test_artifacts/v0/scripts/run_autogluon_tests.sh
@@ -13,5 +13,6 @@ ret=$?
 
 if [ $ret -eq 0 ]
 then
+    echo "This notebook only supports a single GPU setup. Running on multi-GPU systems may lead to unexpected errors"
     jupyter nbconvert --execute --to python docs/tutorials/multimodal/multimodal_prediction/multimodal-quick-start.ipynb
 fi
diff --git a/test/test_artifacts/v1/scripts/run_autogluon_tests.sh b/test/test_artifacts/v1/scripts/run_autogluon_tests.sh
@@ -13,5 +13,6 @@ ret=$?
 
 if [ $ret -eq 0 ]
 then
+    echo "This notebook only supports a single GPU setup. Running on multi-GPU systems may lead to unexpected errors"
     jupyter nbconvert --execute --to python docs/tutorials/multimodal/multimodal_prediction/multimodal-quick-start.ipynb
 fi
diff --git a/test/test_artifacts/v2/scripts/run_autogluon_tests.sh b/test/test_artifacts/v2/scripts/run_autogluon_tests.sh
@@ -13,5 +13,6 @@ ret=$?
 
 if [ $ret -eq 0 ]
 then
+    echo "This notebook only supports a single GPU setup. Running on multi-GPU systems may lead to unexpected errors"
     jupyter nbconvert --execute --to python docs/tutorials/multimodal/multimodal_prediction/multimodal-quick-start.ipynb
 fi