code: def test_mode_phylo(num_leaves, dtype):
      logits, times = random_phylo_logits(num_leaves, dtype)
      d = dist.OneTwoMatching(logits)
      values = d.enumerate_support()
      i = d.log_prob(values).max(0).indices.item()
      expected = values[i]
      actual = d.mode()
      assert_equal(actual, expected)
  
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 15 for calling a global function with example usage: test_mode_phylo(num_leaves, dtype) without return types