code: def test_varying_kwargs():
      def fn(x, scale=1.0):
          return x * scale
  
      jit_fn = pyro.ops.jit.trace(fn)
      x = torch.tensor(1.0)
      for scale in [-1.0, 0.0, 1.0, 10.0]:
          assert_equal(jit_fn(x, scale=scale), fn(x, scale=scale))
  
imports: import torch
  import pyro.ops.jit
  from tests.common import assert_equal
label_desc: Use image node_id 2 for calling a global function with example usage: test_varying_kwargs() and returns: unknown