fix: force download of HF models in tests

.github/workflows/test.yml

@@ -21,8 +21,6 @@ jobs:
           # - "3.12"
     steps:
       - uses: actions/checkout@v4
-      - name: Install Git
-        run: sudo apt-get install git
       - name: Cache Poetry
         uses: actions/cache@v3
         with:

tests/unit/encoders/test_clip.py

@@ -4,8 +4,14 @@ import torch
 from PIL import Image
 
 from semantic_router.encoders import CLIPEncoder
+from transformers import AutoTokenizer, AutoModel
 
 test_model_name = "aurelio-ai/sr-test-clip"
+
+# force the model download
+tokenizer = AutoTokenizer.from_pretrained(test_model_name, force_download=True)
+model = AutoModel.from_pretrained(test_model_name, force_download=True)
+
 clip_encoder = CLIPEncoder(name=test_model_name)
 embed_dim = 64
 

tests/unit/encoders/test_huggingface.py

@@ -4,8 +4,14 @@ import numpy as np
 import pytest
 
 from semantic_router.encoders.huggingface import HuggingFaceEncoder
+from transformers import AutoTokenizer, AutoModel
 
 test_model_name = "aurelio-ai/sr-test-huggingface"
+
+# force the model download
+tokenizer = AutoTokenizer.from_pretrained(test_model_name, force_download=True)
+model = AutoModel.from_pretrained(test_model_name, force_download=True)
+
 encoder = HuggingFaceEncoder(name=test_model_name)