with files attached below I get a python error when launching --data_dir data/mytest index error detected on def construct_A_matrix(self): line A[self.inner_pos, n2_pos] = 1 can you check ?   