code: def test_input_varargs():
      model = nn.Sequential(nn.Linear(1, 1))
      model = Pipe(model, balance=[1], devices=["cpu"])
  
      a = torch.rand(1)
      b = torch.rand(1)
  
      # TypeError: forward() takes 2 positional arguments but 3 were given
      with pytest.raises(TypeError):
          model(a, b)
  
imports: from collections import OrderedDict
  from copy import deepcopy
  import time
  import pytest
  import torch
  from torch import nn
  from fairscale.nn.pipe import Pipe
label_desc: Use image node_id 20 for calling a global function with example usage: test_input_varargs() without return types