code: def test_filter(self):
      init_fn, _ = transform.transform(get_net)
      params = init_fn(jax.random.PRNGKey(428), jnp.ones((1, 1)))
  
      second_layer_params = filtering.filter(
          lambda module_name, *_: module_name == "second_layer", params
      )
      self.assertEqual(
          get_names(second_layer_params),
          {"second_layer/w", "second_layer/b"},
      )
  
      biases = filtering.filter(
          lambda module_name, name, _: name == "b", params
      )  # pytype: disable=wrong-arg-types
      self.assertEqual(
          get_names(biases), {"first_layer/b", "second_layer/b"}
      )
  
imports: import collections
  from collections.abc import Sequence
  import itertools
  import re
  import types
  from typing import Any, Callable
  from absl.testing import absltest
  from absl.testing import parameterized
  from haiku._src import basic
  from haiku._src import data_structures
  from haiku._src import filtering
  from haiku._src import transform
  import jax
  import jax.numpy
label_desc: Use image node_id 9 for calling the FilteringTest obj's underlying member method code with example usage: obj.test_filter() without return types