def cuda_index(self, input_points, extra_data=None): from cuda_indexing import CudaIndexing self.cuda_indexing = CudaIndexing() # we have only one table indexed_data = self.cuda_indexing.batch_indexing( self.uniform_planes[0], input_points) #for data in indexed_data: # self.hash_tables[0].append_val(data, extra_data) # extra_data += 1 self.hash_tables[0].batch_append_vals(indexed_data.tolist(), extra_data)