Skip to content

Conversation

@n1t0
Copy link
Contributor

@n1t0 n1t0 commented Jan 15, 2021

The extras["retrieval"] is defined a few lines above as:

transformers/setup.py

Lines 217 to 222 in 28b2601

if os.name == "nt": # windows
extras["retrieval"] = deps_list("datasets") # faiss is not supported on windows
extras["flax"] = [] # jax is not supported on windows
else:
extras["retrieval"] = deps_list("faiss-cpu", "datasets")
extras["flax"] = deps_list("jax", "jaxlib", "flax")

and then it seems to be overridden just below, probably linking to faiss-cpu being included even on windows.

This PR removes the second assignment.

cc @LysandreJik @sgugger @stas00

Copy link
Collaborator

@sgugger sgugger left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Thanks for fixing!

Copy link
Member

@LysandreJik LysandreJik left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

LGTM, thanks @n1t0!

@LysandreJik LysandreJik merged commit 72fc9ab into master Jan 18, 2021
@LysandreJik LysandreJik deleted the fix-retrieval-deps branch January 18, 2021 09:24
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.

4 participants