code: def is_python_scalar(x: Any) -> bool:
      try:
          return x.aval.weak_type and np.ndim(x) == 0
      except AttributeError:
          return type(x) in python_scalar_dtypes
  
imports: from __future__ import annotations
  import abc
  import builtins
  import functools
  from typing import cast, overload, Any, Literal, Union
  import warnings
  import ml_dtypes
  import numpy
  from jax._src import config
  from jax._src.typing import DType, DTypeLike
  from jax._src.util import set_module
  from jax._src import traceback_util
label_desc: Use image node_id 21 for calling a global function with example usage: is_python_scalar(x) and returns: unknown, unknown