modified unittest

bob/learn/pytorch/test/test.py

@@ -225,7 +225,8 @@ def test_MCCNNtrainer():
   from ..architectures import MCCNN
   net = MCCNN(num_channels=4)
 
-  dataloader = torch.utils.data.DataLoader(DummyDataSetMCCNN(), batch_size=32, shuffle=True)
+ dataloader={}
+ dataloader{'train'} = torch.utils.data.DataLoader(DummyDataSetMCCNN(), batch_size=32, shuffle=True)
   
   from ..trainers import MCCNNTrainer
   trainer = MCCNNTrainer(net, verbosity_level=3)