From 6bddd964ded5237240f736499735853ce6c5581f Mon Sep 17 00:00:00 2001
From: Maciej Kilian <iejmac@ip-26-0-155-200.us-west-2.compute.internal>
Date: Tue, 24 Oct 2023 04:00:59 +0000
Subject: [PATCH] fix tests

---
 tests/test_modules.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/tests/test_modules.py b/tests/test_modules.py
index e18b255..fb762d0 100644
--- a/tests/test_modules.py
+++ b/tests/test_modules.py
@@ -86,7 +86,7 @@ def test_writer(writer_type):
         vid_embeds = [np.ones((N_FRAMES, lat_dim), dtype=float) * i for i in range(N_VIDS)]
 
         for i, emb in enumerate(vid_embeds):
-            fake_metadata = {"caption": str(i), "x": i}
+            fake_metadata = {"json": {"caption": str(i), "x": i}, "txt": str(i)}
             writer.write(emb, str(i), fake_metadata)
         writer.close()