Update model.py

src/models/model.py

@@ -131,3 +131,8 @@ class PLDataModule(LightningDataModule):
         return DataLoader(
             self.train_dataset, batch_size=self.batch_size, shuffle=True, num_workers=2
         )
+
+    def test_dataloader(self):
+        """ test dataloader """
+        return DataLoader(
+            self.test_dataset, batch_size=self.batch_size, shuffle=False, num_workers=2