code: def compute_optimal_action_with_environment_dynamics(
      observation, environment_dynamics
  ):
      """Computes the optimal action using the environment dynamics.
  
      Args:
        observation: a (possibly batched) observation.
        environment_dynamics: environment dynamics object (an instance of
          `non_stationary_stochastic_environment.EnvironmentDynamics`)
  
      Returns:
        The optimal action.
      """
      return environment_dynamics.compute_optimal_action(observation)
  
imports: from __future__ import absolute_import
  from __future__ import division
  from __future__ import print_function
  import functools
  import gin
  import numpy
  from six.moves import range
  import tensorflow
  from tf_agents.bandits.agents import utils
  from tf_agents.bandits.environments import wheel_py_environment
label_desc: Use image node_id 14 for calling a global function with example usage: compute_optimal_action_with_environment_dynamics(observation, environment_dynamics) and returns: environment_dynamics