code: def to_complex_dtype(dtype: DTypeLike) -> DType:
      ftype = to_inexact_dtype(dtype)
      if ftype in [np.dtype("float64"), np.dtype("complex128")]:
          return np.dtype("complex128")
      return np.dtype("complex64")
  
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 4 for calling a global function with example usage: to_complex_dtype(dtype) and returns: np, np