code: def _warn_if_unsafe(window_shape, strides):
      unsafe = lambda size: isinstance(size, int) and size != 1
      if unsafe(window_shape) or unsafe(strides):
          warnings.warn(
              _VMAP_SHAPE_INFERENCE_WARNING, DeprecationWarning
          )
  
imports: from collections.abc import Sequence
  from typing import Optional, Union
  import warnings
  from haiku._src import module
  import jax
  from jax import lax
  import jax.numpy
  import numpy
label_desc: Use image node_id 2 for calling a global function with example usage: _warn_if_unsafe(window_shape, strides) without return types