code: def make_distrib_ff_q_func(env):
      n_atoms = 51
      v_max = 10
      v_min = -10
      return chainerrl.q_functions.DistributionalFCStateQFunctionWithDiscreteAction(  # NOQA
          env.observation_space.low.size,
          env.action_space.n,
          n_atoms=n_atoms,
          v_min=v_min,
          v_max=v_max,
          n_hidden_channels=20,
          n_hidden_layers=1,
      )
  
imports: import unittest
  import chainer
  import chainer.links
  from chainer import testing
  import numpy
  import basetest_dqn_like
  from basetest_training import _TestBatchTrainingMixin
  import chainerrl
  from chainerrl.agents import categorical_dqn
  from chainerrl.agents.categorical_dqn import compute_value_loss
  from chainerrl.agents.categorical_dqn import compute_weighted_value_loss
  from chainerrl.agents import CategoricalDQN
label_desc: Use image node_id 2 for calling a global function with example usage: make_distrib_ff_q_func(env) and returns: chainerrl