def METHOD_NAME(*args, **kwargs):
    if "param_dtype" in kwargs:
        err_msg = f"""
                *************************************************************************
                Use `flax.linen.{module_name}` instead of `netket.nn.{module_name}`.
                *************************************************************************
        You are specifying `param_dtype` so you should be good.
        """
        raise KeyError(dedent(err_msg))
    else:
        warnings.warn(
            _warning_string(module_name), category=FutureWarning, stacklevel=3
        )
    if "dtype" in kwargs:
        dtype = kwargs.pop("dtype")
        return original_module(*args, param_dtype=dtype, **kwargs)
    else:
        return original_module(*args, **kwargs)