code: def _optional_scalar_param(
      param: Optional[Union[float, int]]
  ) -> Optional[Union[float, int]]:
      return param
  
imports: from collections.abc import Callable
  import os
  from typing import Any, Optional, Union
  from absl.testing import absltest
  from absl.testing import parameterized
  from etils import epath
  from lit_nlp.api import dtypes
  from lit_nlp.api import types
  import numpy
label_desc: Use image node_id 16 for calling a global function with example usage: _optional_scalar_param(param) and returns: param