Installing `datasets` on M1 computers

[david1542]

Describe the bug

I wanted to install datasets dependencies on my M1 (in order to start contributing to the project). However, I got an error regarding tensorflow.

On M1, tensorflow-macos needs to be installed instead. Can we add a conditional requirement, so that tensorflow-macos would be installed on M1?

Steps to reproduce the bug

Fresh clone this project (on m1), create a virtualenv and run this:

pip install -e ".[dev]"

Expected results

Installation should be smooth, and all the dependencies should be installed on M1.

Actual results

You should receive an error, saying pip couldn't find a version that matches this pattern:

tensorflow>=2.3,!=2.6.0,!=2.6.1

Environment info

• datasets version: 2.6.2.dev0
• Platform: macOS-12.6-arm64-arm-64bit
• Python version: 3.9.6
• PyArrow version: 7.0.0
• Pandas version: 1.5.0

[albertvillanova]

Thanks for reporting, @david1542.