From da8839f2789e332e688fc36521d41f61a8efaca3 Mon Sep 17 00:00:00 2001
From: Travis Addair <tgaddair@gmail.com>
Date: Wed, 13 Mar 2024 09:39:13 -0700
Subject: [PATCH] Mock

---
 server/tests/utils/test_lora.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/server/tests/utils/test_lora.py b/server/tests/utils/test_lora.py
index 8675b825d..7f442a68a 100644
--- 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],