test-fix

logix-project · Jan 28, 2024 · 79d94c2 · 79d94c2
1 parent a8eac07
commit 79d94c2
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 10 deletions.
diff --git a/analog/test/config.yaml b/analog/test/config.yaml
@@ -11,7 +11,7 @@ logging:
   log_dtype: none
   num_workers: 1
 lora:
-  init: random
+  init: pca
   parameter_sharing: false
   parameter_sharing_groups: null
   rank: 64

diff --git a/tests/storage/test_util_integration.py b/tests/storage/test_util_integration.py
@@ -48,11 +48,10 @@ def tearDownClass(cls):
         shutil.rmtree(cls.test_dir)
 
     def test_write_and_read(self):
-        data_buffer = [
-            (i, {"dummy_data": arr}) for i, arr in enumerate(generate_random_arrays())
-        ]
+        rand_arr = generate_random_arrays()
+        data_buffer = [(i, {"dummy_data": arr}) for i, arr in enumerate(rand_arr)]
         filename = "test_data"
-        order = ["dummy_data"]
+        order = [["dummy_data"]]
         MemoryMapHandler.write(self.test_dir, filename, data_buffer, order)
 
         mmap = None
@@ -78,12 +77,10 @@ def test_read(self):
         expected_files_path = os.path.join(
             os.path.dirname(os.path.abspath(__file__)), "test_mmap_data"
         )
-
+        static_arr = generate_static_arrays()
         filename = "test_data"
-        data_buffer = [
-            (i, {"dummy_data": arr}) for i, arr in enumerate(generate_static_arrays())
-        ]
-        order = ["dummy_data"]
+        data_buffer = [(i, {"dummy_data": arr}) for i, arr in enumerate(static_arr)]
+        order = [["dummy_data"]]
         MemoryMapHandler.write(expected_files_path, filename, data_buffer, order)
         mmap = None
         with MemoryMapHandler.read(expected_files_path, filename) as mm: