code: def test_mode(num_nodes, dtype):
      logits = torch.randn(num_nodes, num_nodes, dtype=dtype) * 10
      d = dist.OneOneMatching(logits)
      values = d.enumerate_support()
      i = d.log_prob(values).max(0).indices.item()
      expected = values[i]
      actual = d.mode()
      assert_equal(actual, expected)
      assert (actual == expected).all()
  
file_path: pyro/tests/distributions/test_one_one_matching.py
imports: import logging
  import math
  import pytest
  import torch
  import pyro.distributions
  from tests.common import assert_close, assert_equal, xfail_if_not_implemented
label_desc: Use image node_id 10 for calling a global function with example usage: test_mode(num_nodes, dtype) without return types