Did you intentionally not use ```np.non_zero()``` here? Are you trying to filter out negative weights? Am I reading too much into this :wink:?