Error with first provided example

[anonym-ai]

When running the example in the first notebook,
I have the following error:

in code:

        slice_sizes = x_encoded.sum(axis=0).A[0]
        slice_errors = errors @ x_encoded
        max_slice_errors = x_encoded.T.multiply(errors).max(axis=1).A[:, 0]

AttributeError: coo_matrix has no attribute A

It seems like x_encoded.T.multiply(errors).max(axis=1) is a scipy.sparse.coo_matrix and not a np.matrix

I have no idea if x_encoded.T.multiply(errors) is as scipy.sparse.csr like matrix

but modifing the line with:

-        max_slice_errors = x_encoded.T.multiply(errors).max(axis=1).A[:, 0]
+        max_slice_errors = x_encoded.T.multiply(errors).max(axis=1).toarray()[:, 0]

The whole code runs smoothly.
I guess this is not optimal.