Fix milvus test Error: 'NameError: name 'MockConfig' is not defined'

Significant-Gravitas · Apr 17, 2023 · 1d4dc0c · 1d4dc0c
1 parent 6fb1369
commit 1d4dc0c
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/milvus_memory_test.py b/tests/milvus_memory_test.py
@@ -26,7 +26,7 @@ class TestMilvusMemory(unittest.TestCase):
 
         def setUp(self) -> None:
             """Set up the test environment"""
-            self.cfg = MockConfig()
+            self.cfg = mock_config()
             self.memory = MilvusMemory(self.cfg)
 
         def test_add(self) -> None: