TypeError when contracting with diagonal view

[sunqm]

Running the following contraction using pytblis.contract raises a TypeError:

import numpy as np
import pytblis

pytblis.contract('ia,a->ia', np.eye(3), np.eye(3).diagonal())

TypeError: mult(): incompatible function arguments. The following argument types are supported:
    1. mult(A: ndarray[], B: ndarray[], C: ndarray[], idx_A: str, idx_B: str, idx_C: str, alpha: complex = 1.0, beta: complex = 0.0, conja: bool = False, conjb: bool = False)
 -> None

Invoked with types: ndarray, ndarray, ndarray, str, str, str, kwargs = { alpha: complex, beta: complex, conja: bool, conjb: bool }

[chillenb]

I think this occurs because the .diagonal() function returns a non-writable view. Will fix tomorrow.