Fix StaticVectors after floret+mypy merge (#9566)

This commit is contained in:
Adriane Boyd 2021-10-29 16:25:43 +02:00 committed by GitHub
parent 322635e371
commit bb26550e22
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -77,7 +77,7 @@ def forward(
model.inc_grad(
"W",
model.ops.gemm(
cast(Floats2d, d_output.data), model.ops.as_contig(V[rows]), trans1=True
cast(Floats2d, d_output.data), model.ops.as_contig(V), trans1=True
),
)
return []