Gagan Bhatia commited on
Commit
3b5d713
1 Parent(s): 2c57b14

Update model.py

Browse files
Files changed (1) hide show
  1. src/models/model.py +4 -1
src/models/model.py CHANGED
@@ -143,7 +143,10 @@ class PLDataModule(LightningDataModule):
143
  def test_dataloader(self):
144
  """test dataloader"""
145
  return DataLoader(
146
- self.test_dataset, batch_size=self.batch_size, shuffle=False, num_workers=self.num_workers
 
 
 
147
  )
148
 
149
  def val_dataloader(self):
 
143
  def test_dataloader(self):
144
  """test dataloader"""
145
  return DataLoader(
146
+ self.test_dataset,
147
+ batch_size=self.batch_size,
148
+ shuffle=False,
149
+ num_workers=self.num_workers,
150
  )
151
 
152
  def val_dataloader(self):