code: def test_custom_init():
      with pytest.raises(TypeError):
          eqx.nn.Linear(3, 4)  # pyright: ignore
  
      with pytest.raises(TypeError):
          eqx.nn.Linear(3)  # pyright: ignore
  
      with pytest.raises(TypeError):
          eqx.nn.Linear(out_features=4)
  
imports: import warnings
  from typing import Union
  import equinox
  import jax
  import jax.nn
  import jax.numpy
  import jax.random
  import pytest
label_desc: Use image node_id 1 for calling a global function with example usage: test_custom_init() without return types