code: def constant(step_size) -> Schedule:
      def schedule(i):
          return step_size
  
      return schedule
  
imports: from __future__ import annotations
  from typing import Any, Callable, NamedTuple
  from collections import namedtuple
  import functools
  from functools import partial
  import jax.numpy
  from jax._src.util import safe_zip, safe_map, unzip2
  from jax import tree_util
  from jax.tree_util import tree_map, tree_flatten, tree_unflatten, register_pytree_node
label_desc: Use image node_id 11 for calling a global function with example usage: constant(step_size) and returns: schedule, step_size