code: def check_for_prngkeys(fun_name: str, *args: Any):
      """Check if args don't match and none of the args have typed prng dtype"""
      arg_dtypes = [dtypes.dtype(arg) for arg in args]
      if len(set(arg_dtypes)) < 2:
          return  # Will be caught by extended dtype impl rules.
      if any(
          dtypes.issubdtype(dt, dtypes.prng_key) for dt in arg_dtypes
      ):
          if len(arg_dtypes) == 1:
              raise TypeError(
                  f"{fun_name} does not accept dtype {str(arg_dtypes[0])}."
              )
          else:
              raise TypeError(
                  f"{fun_name} does not accept dtypes {', '.join(map(str, arg_dtypes))}."
              )
  
imports: from __future__ import annotations
  from collections.abc import Sequence
  from functools import partial
  import re
  import textwrap
  from typing import Any, Callable, NamedTuple, TypeVar
  import warnings
  from jax._src import api
  from jax._src import config
  from jax._src import core
  from jax._src import dtypes
  from jax._src.lax import lax
  from jax._src.util import safe_zip, safe_map
  from jax._src.typing import Array, ArrayLike, DimSize, DType, DTypeLike, Shape
  import numpy
label_desc: Use image node_id 16 for calling a global function with example usage: check_for_prngkeys(fun_name) without return types