code: def setUp(self):
      torch.manual_seed(42)
      self.pytorch_network = nn.Linear(1, 1)
      self.loss_function = nn.MSELoss()
      self.optimizer = torch.optim.SGD(
          self.pytorch_network.parameters(), lr=1e-3
      )
      self.model = Model(
          self.pytorch_network, self.optimizer, self.loss_function
      )
      self.mock_callback = MagicMock(spec=Callback)
      self.delay_callback = DelayCallback(self.mock_callback)
      self.train_dict = {"loss": ANY, "time": ANY}
      self.log_dict = {"loss": ANY, "val_loss": ANY, "time": ANY}
  
file_path: poutyne/tests/framework/callbacks/test_delay.py
imports: import unittest
  from unittest import TestCase
  from unittest.mock import ANY, MagicMock, call
  import torch
  import torch.nn
  from poutyne import Callback, CallbackList, DelayCallback, Model
  from tests.framework.tools import some_data_generator
label_desc: Use image node_id 1 for calling the DelayCallbackTest obj's underlying member method code with example usage: obj.setUp() without return types