code: def net(self, net_cls, module_cls):
      return net_cls(
          module_cls,
          max_epochs=20,
          lr=0.1,
      )
  
imports: from functools import partial
  import numpy
  import pytest
  from sklearn.base import clone
  from skorch.tests.conftest import INFERENCE_METHODS
label_desc: Use image node_id 5 for calling the TestNeuralNetRegressor obj's underlying member method code with example usage: obj.net(net_cls, module_cls) and returns: net_cls