Fix test import

Signed-off-by: Rafael Vasquez <rafvasq21@gmail.com>
opendatahub-io · Aug 1, 2024 · b55edac · b55edac
1 parent 8637849
commit b55edac
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/tests/test_hub.py b/tests/test_hub.py
@@ -2,7 +2,8 @@
 
 import pytest
 from huggingface_hub.utils import LocalEntryNotFoundError
-from tgis_utils.hub import (
+
+from vllm_tgis_adapter.tgis_utils.hub import (
     convert_files,
     download_weights,
     weight_files,