Mock

predibase · Mar 13, 2024 · da8839f · da8839f
1 parent 16ea686
commit da8839f
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/server/tests/utils/test_lora.py b/server/tests/utils/test_lora.py
@@ -1,4 +1,5 @@
 from typing import List
+from unittest import mock
 import pytest
 
 import torch
@@ -8,6 +9,7 @@
 from lorax_server.utils.sgmv import MIN_RANK_CUSTOM
 
 
+@mock.patch("lorax_server.utils.lora.get_tmp_tensors", return_value=(torch.empty(0), torch.empty(0)))
 @pytest.mark.parametrize("lora_ranks", [
     [8, 16],
     [32, 64],