Skip to content

Conversation

@lhoestq
Copy link
Member

@lhoestq lhoestq commented Aug 11, 2020

When getting items using a column name as a key, numpy arrays were not stacked.
I fixed that and added some tests.

There is another issue that still needs to be fixed though: when getting items using a column name as a key, pytorch tensors are not stacked (it outputs a list of tensors). This PR should help with the to fix this issue.

@lhoestq lhoestq merged commit 427216e into master Aug 11, 2020
@lhoestq lhoestq deleted the fix-numpy-stacking branch August 11, 2020 13:49
@lhoestq
Copy link
Member Author

lhoestq commented Aug 11, 2020

This PR also fixed a bug where numpy arrays were returned instead of pytorch tensors when getting with a clumn as a key.

@lhoestq lhoestq mentioned this pull request Aug 12, 2020
vegarab pushed a commit to vegarab/nlp that referenced this pull request Aug 18, 2020
* fix numpy stacking when getting with a column as a key

* style
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.

2 participants