From 4362a884280c493a7ea138cd043f8297f9fab346 Mon Sep 17 00:00:00 2001
From: Steven Silvester <steve.silvester@mongodb.com>
Date: Mon, 16 Jun 2025 10:34:48 -0500
Subject: [PATCH 1/2] Revert "INTPYTHON-640 Temporarily skip docarray tests
 until proper investigation. (#77)"

This reverts commit 5f2964dcecb4f21975d2660b9d720e9cf29bf72b.
---
 .evergreen/config.yml | 46 +++++++++++++++++++++----------------------
 1 file changed, 22 insertions(+), 24 deletions(-)

diff --git a/.evergreen/config.yml b/.evergreen/config.yml
index 9ca7ea5..c13402a 100644
--- a/.evergreen/config.yml
+++ b/.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

From a1f3d64d247107e0be04b54546f32bb9fd3c45a4 Mon Sep 17 00:00:00 2001
From: Steven Silvester <steve.silvester@mongodb.com>
Date: Mon, 16 Jun 2025 10:35:43 -0500
Subject: [PATCH 2/2] INTPYTHON-640 Add retry for docarray tests

---
 docarray/run.sh | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/docarray/run.sh b/docarray/run.sh
index 36d4433..42decd4 100644
--- a/docarray/run.sh
+++ b/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