pythonPackages.pytorch: fix weirdly broken wheel version number

Otherwise, the wheel gets built with invalid metadata -- causing
'torch >= 1.0.0' to be unsatisfiable in other python packages, for
instance.

Signed-off-by: Austin Seipp <aseipp@pobox.com>
This commit is contained in:
Austin Seipp 2018-12-24 05:46:51 -06:00
parent be96560327
commit c6ecb3887f

View file

@ -58,6 +58,12 @@ in buildPythonPackage rec {
done
'';
# Override the (weirdly) wrong version set by default. See
# https://github.com/NixOS/nixpkgs/pull/52437#issuecomment-449718038
# https://github.com/pytorch/pytorch/blob/v1.0.0/setup.py#L267
PYTORCH_BUILD_VERSION = version;
PYTORCH_BUILD_NUMBER = 0;
buildInputs = [
cmake
numpy.blas