Add missing milvus mark for two tests which require milvus

nv-morpheus · Nov 16, 2023 · 6808acb · 6808acb
1 parent 9ec6d7e
commit 6808acb
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/tests/llm/test_rag_standalone_pipe.py b/tests/llm/test_rag_standalone_pipe.py
@@ -208,6 +208,7 @@ def test_rag_standalone_pipe_openai(config: Config,
 
 @pytest.mark.usefixtures("nemollm")
 @pytest.mark.usefixtures("ngc_api_key")
+@pytest.mark.milvus
 @pytest.mark.use_python
 @pytest.mark.use_cudf
 @pytest.mark.parametrize("repeat_count", [5])
@@ -239,6 +240,7 @@ def test_rag_standalone_pipe_integration_nemo(config: Config,
 
 @pytest.mark.usefixtures("openai")
 @pytest.mark.usefixtures("openai_api_key")
+@pytest.mark.milvus
 @pytest.mark.use_python
 @pytest.mark.use_cudf
 @pytest.mark.parametrize("repeat_count", [5])