code: def test_module_smoke(
      module, batch_size, expected_shapes, expected_dtypes
  ):
      batch = module.describe_inputs(batch_size).zeros()
      outputs = module(**batch)
      assert_shapes_and_dtypes(
          outputs, expected_shapes, expected_dtypes
      )
  
file_path: gluonts/test/torch/model/test_modules.py
imports: import pytest
  import torch
  from gluonts.torch.distributions import QuantileOutput
  from gluonts.torch.model.deepar import DeepARModel
  from gluonts.torch.model.mqf2 import MQF2MultiHorizonModel
  from gluonts.torch.model.simple_feedforward import SimpleFeedForwardModel
  from gluonts.torch.model.tft import TemporalFusionTransformerModel
label_desc: Use image node_id 2 for calling a global function with example usage: test_module_smoke(module, batch_size, expected_shapes, expected_dtypes) without return types