class_docstr: Tests for Scenic Optax utils.
class_docstr_tok: ["Tests","for","Scenic","Optax","utils","."]
docstr: Tests `make_schedule` with None schedule.
docstr_tok: ["Tests","`","make_schedule","`","with","None","schedule","."]
code: def test_make_schedule_with_none(self):
      """Tests `make_schedule` with None schedule."""
  
      def lr_fn(cfg: ml_collections.ConfigDict) -> Optional[float]:
          """Dummy LR schedule creation function."""
          if cfg.lr_configs is None:
              return None
          return float(cfg.lr_configs.lr)
  
      new_schedule = optax.make_schedule(None, lr_fn)
      target_schedule = [("(.*)", "all", (None, None))]
      self.assertAllEqual(target_schedule, new_schedule)
  
imports: from typing import Optional
  from absl.testing import absltest
  from absl.testing import parameterized
  import ml_collections
  from scenic.train_lib import optax
  import tensorflow
label_desc: Use image node_id 10 for calling the OptaxTest obj's underlying member method code with example usage: obj.test_make_schedule_with_none() and returns: float, None