From f5eeb1c3dc2a69199fb465a35c03901b05e0da8d Mon Sep 17 00:00:00 2001
From: James Briggs <james.briggs@hotmail.com>
Date: Sat, 2 Mar 2024 16:19:08 +0800
Subject: [PATCH] add mock model for huggingface test

---
 coverage.xml                            | 2 +-
 tests/unit/encoders/test_huggingface.py | 3 ++-
 2 files changed, 3 insertions(+), 2 deletions(-)

diff --git a/coverage.xml b/coverage.xml
index fefa412d..2144ec09 100644
--- a/coverage.xml
+++ b/coverage.xml
@@ -1,5 +1,5 @@
 <?xml version="1.0" ?>
-<coverage version="7.4.3" timestamp="1709366989240" lines-valid="2072" lines-covered="1619" line-rate="0.7814" branches-covered="0" branches-valid="0" branch-rate="0" complexity="0">
+<coverage version="7.4.3" timestamp="1709367522085" lines-valid="2072" lines-covered="1619" line-rate="0.7814" branches-covered="0" branches-valid="0" branch-rate="0" complexity="0">
 	<!-- Generated by coverage.py: https://coverage.readthedocs.io/en/7.4.3 -->
 	<!-- Based on https://raw.githubusercontent.com/cobertura/web/master/htdocs/xml/coverage-04.dtd -->
 	<sources>
diff --git a/tests/unit/encoders/test_huggingface.py b/tests/unit/encoders/test_huggingface.py
index 1e916f3e..7c17f98f 100644
--- a/tests/unit/encoders/test_huggingface.py
+++ b/tests/unit/encoders/test_huggingface.py
@@ -5,7 +5,8 @@ import pytest
 
 from semantic_router.encoders.huggingface import HuggingFaceEncoder
 
-encoder = HuggingFaceEncoder()
+test_model_name = "hf-internal-testing/tiny-bert"
+encoder = HuggingFaceEncoder(name=test_model_name)
 
 
 class TestHuggingFaceEncoder:
-- 
GitLab