Skip to content

Conversation

@gsakkis
Copy link
Contributor

@gsakkis gsakkis commented Aug 22, 2022

Update reader tests to go through the _csr_to_coo_collate collator.

Currently this collator is not being tested because the default tensor kind for 2D integer sparse arrays on PyTorchTileDBDataLoader is SPARSE_CSR so we have to pass expicitly tensor_kind=TensorKind.SPARSE_COO to ArrayParams to select conversion to COO tensors. In this case PyTorchTileDBDataLoader invokes the _csr_to_coo_collate.

@gsakkis gsakkis requested a review from georgeSkoumas August 22, 2022 08:22
Base automatically changed from gsa/bump-pytorch to master August 22, 2022 13:29
@gsakkis gsakkis merged commit ab539f4 into master Aug 22, 2022
@gsakkis gsakkis deleted the gsa/test_csr_to_coo_collate branch August 22, 2022 13:32
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

3 participants