code: def output_dir(tmp_path):
      _output_dir = tmp_path / "out"
      _output_dir.mkdir()
      model = torch.nn.Sequential(
          torch.nn.Conv2d(1, 1, 3),
          torch.nn.BatchNorm2d(1),
          torch.nn.Linear(1, 1),
          torch.nn.LSTM(1, 1),
          torch.nn.LayerNorm(1),
      )
      torch.save(model.state_dict(), _output_dir / "1epoch.pth")
      torch.save(model.state_dict(), _output_dir / "2epoch.pth")
      torch.save(model.state_dict(), _output_dir / "3epoch.pth")
      return _output_dir
  
imports: import pytest
  import torch
  from espnet2.main_funcs.average_nbest_models import average_nbest_models
  from espnet2.train.reporter import Reporter
label_desc: Use image node_id 2 for calling a global function with example usage: output_dir(tmp_path) and returns: _output_dir