Skip to content

Commit

Permalink
Fix staging CI setup (#1229)
Browse files Browse the repository at this point in the history
- [x] add `time.sleep` in `DummyLLM` to make it run longer for the resue
test cases.
- [x] always cache even the test is failed. 
- [x] skip pinecone tests due to free-tier limit.
  • Loading branch information
xzdandy authored Sep 27, 2023
1 parent c65b344 commit 645a8f2
Show file tree
Hide file tree
Showing 3 changed files with 5 additions and 1 deletion.
1 change: 1 addition & 0 deletions .circleci/config.yml
Original file line number Diff line number Diff line change
Expand Up @@ -253,6 +253,7 @@ jobs:
if [[ $PY_VERSION = "3.10" ]] || [[ $PY_VERSION = "3.11" ]]; then
export SETUPTOOLS_USE_DISTUTILS=stdlib
fi
set +e # To make sure the later cache step is not skipped.
bash script/test/test.sh -m "<< parameters.mode >>"
# Enable cache save conditionally (to avoid empty cache in Notebooks)
Expand Down
1 change: 1 addition & 0 deletions test/integration_tests/long/test_similarity.py
Original file line number Diff line number Diff line change
Expand Up @@ -463,6 +463,7 @@ def test_end_to_end_index_scan_should_work_correctly_on_image_dataset_chromadb(
drop_query = "DROP INDEX testChromaDBIndexImageDataset"
execute_query_fetch_all(self.evadb, drop_query)

@pytest.mark.skip(reason="require pinecone")
@pinecone_skip_marker
def test_end_to_end_index_scan_should_work_correctly_on_image_dataset_pinecone(
self,
Expand Down
4 changes: 3 additions & 1 deletion test/util.py
Original file line number Diff line number Diff line change
Expand Up @@ -17,6 +17,7 @@
import os
import shutil
import socket
import time
from contextlib import closing
from itertools import repeat
from multiprocessing import Pool
Expand Down Expand Up @@ -713,5 +714,6 @@ def forward(self, text_df):
results.append(("" if prompt is None else prompt) + query + " " + content)

df = pd.DataFrame({"response": results})

# Make it slower
time.sleep(1)
return df

0 comments on commit 645a8f2

Please sign in to comment.