INTPYTHON-640 Add retry for docarray tests (#80)

Author: blink1073

.evergreen/config.yml

@@ -180,20 +180,19 @@ tasks:
       - func: "setup remote atlas"
       - func: "execute tests"
 
-# TODO https://jira.mongodb.org/browse/INTPYTHON-640
-#  - name: test-docarray-local
-#    tags: [local]
-#    commands:
-#      - func: "fetch repo"
-#      - func: "setup local atlas"
-#      - func: "execute tests"
-# TODO https://jira.mongodb.org/browse/INTPYTHON-640
-#  - name: test-docarray-remote
-#    tags: [remote]
-#    commands:
-#      - func: "fetch repo"
-#      - func: "setup remote atlas"
-#      - func: "execute tests"
+  - name: test-docarray-local
+    tags: [local]
+    commands:
+      - func: "fetch repo"
+      - func: "setup local atlas"
+      - func: "execute tests"
+
+  - name: test-docarray-remote
+    tags: [remote]
+    commands:
+      - func: "fetch repo"
+      - func: "setup remote atlas"
+      - func: "execute tests"
 
   - name: test-pymongo-voyageai-local
     tags: [local]
@@ -330,16 +329,15 @@ buildvariants:
       # - name: test-llama-index-remote
       #   batchtime: 10080  # 1 week
 
-# TODO https://jira.mongodb.org/browse/INTPYTHON-640
-#  - name: test-docarray-rhel
-#    display_name: DocArray RHEL
-#    expansions:
-#      DIR: docarray
-#    run_on:
-#      - rhel87-small
-#    tasks:
-#      - name: test-docarray-local
-#      - name: test-docarray-remote
+  - name: test-docarray-rhel
+    display_name: DocArray RHEL
+    expansions:
+      DIR: docarray
+    run_on:
+      - rhel87-small
+    tasks:
+      - name: test-docarray-local
+      - name: test-docarray-remote
 
   - name: test-pymongo-voyageai-rhel
     display_name: PyMongo-VoyageAI RHEL

docarray/run.sh

@@ -30,6 +30,6 @@ set +x
 
 # Run tests. Sensitive variables in Evergreen come from Evergeen project: ai-ml-pipeline-testing/
 # shellcheck disable=SC2154
-MONGODB_URI="$MONGODB_URI" \
-MONGODB_DATABASE="docarray_test_db" \
-pytest -v tests/index/mongo_atlas
+export MONGODB_URI="$MONGODB_URI"
+export MONGODB_DATABASE="docarray_test_db"
+pytest -v tests/index/mongo_atlas || pytest --lf