formatting

This commit is contained in:
0xArty
2023-04-14 20:42:28 +01:00
parent a0e3c238a4
commit 328ba5e69e
40 changed files with 911 additions and 592 deletions

View File

@@ -5,16 +5,19 @@ from autogpt.memory.local import LocalCache
def MockConfig():
return type('MockConfig', (object,), {
'debug_mode': False,
'continuous_mode': False,
'speak_mode': False,
'memory_index': 'auto-gpt',
})
return type(
"MockConfig",
(object,),
{
"debug_mode": False,
"continuous_mode": False,
"speak_mode": False,
"memory_index": "auto-gpt",
},
)
class TestLocalCache(unittest.TestCase):
def setUp(self):
self.cfg = MockConfig()
self.cache = LocalCache(self.cfg)
@@ -49,5 +52,5 @@ class TestLocalCache(unittest.TestCase):
self.assertEqual(stats, (1, self.cache.data.embeddings.shape))
if __name__ == '__main__':
if __name__ == "__main__":
unittest.main()