Enable embeddings tests. (#67)

These now work due to upstream changes that were pulled in.

Author: tdoublep

.github/workflows/test-spyre.yml

@@ -20,10 +20,13 @@ jobs:
     - name: Run Spyre tests within docker container
       run: |
         docker run -i --rm --entrypoint /bin/bash vllm-spyre -c '''
+          pip install pytest sentence-transformers && \
           python3.12 -c "from transformers import pipeline; pipeline(\"text-generation\", model=\"JackFram/llama-160m\")" && \
           export VARIANT=$(ls /root/.cache/huggingface/hub/models--JackFram--llama-160m/snapshots/) && \
           mkdir -p /models && \
           ln -s /root/.cache/huggingface/hub/models--JackFram--llama-160m/snapshots/${VARIANT} /models/llama-194m && \
-          pip install pytest sentence-transformers && \
+          python3.12 -c "from sentence_transformers import SentenceTransformer; SentenceTransformer(\"sentence-transformers/all-roberta-large-v1\")" && \
+          export VARIANT=$(ls /root/.cache/huggingface/hub/models--sentence-transformers--all-roberta-large-v1/snapshots/) && \
+          ln -s /root/.cache/huggingface/hub/models--sentence-transformers--all-roberta-large-v1/snapshots/${VARIANT} /models/all-roberta-large-v1 && \
           python3.12 -m pytest tests/spyre -v
         '''

tests/spyre/test_spyre_embeddings.py

@@ -10,7 +10,6 @@
                         st_embeddings)
 
 
-@pytest.mark.skip("Skip until failure is resolved.")
 @pytest.mark.parametrize("model", ["/models/all-roberta-large-v1"])
 @pytest.mark.parametrize("prompts", [[
     "The capital of France is Paris."