Currently used a hacky fix which keeps the original `val_map` unchanged, and pass `val_map | {"np": np}` to the shape analysis pass.
Currently used a hacky fix which keeps the original
val_mapunchanged, and passval_map | {"np": np}to the shape analysis pass.