fix: unit tests

Signed-off-by: Avik Basu <ab93@users.noreply.github.com>
numaproj · Aug 11, 2023 · 2c458aa · 2c458aa
1 parent 8daae0b
commit 2c458aa
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/tools/test_data.py b/tests/tools/test_data.py
@@ -116,7 +116,7 @@ def test_dataloader(self):
             StreamingDataLoader(
                 self.data,
                 seq_len=SEQ_LEN,
-                dataloader=DataLoader(StreamingDataset(self.data, seq_len=SEQ_LEN)),
+                dataset=DataLoader(StreamingDataset(self.data, seq_len=SEQ_LEN)),
             )