from beartype import beartype
from beartype.vale import IsDependent
import numpy as np

@beartype
def spam(first: np.ndarray, second: np.ndarray) -> Annotated[
    np.ndarray, IsDependent[lambda result, first, second: 
        result.shape == (first.shape[0], second.shape[1])]]:
    """Matrix multiplication"""
    return a @ b

thread 'main' panicked at 'The symbol must be present in the symbol table, even when it is undefined in python.', compiler/codegen/src/compile.rs:478:57
note: run with `RUST_BACKTRACE=1` environment variable to display a backtrace