diff options
Diffstat (limited to 'pkgs/development/python-modules/pytorch/default.nix')
| -rw-r--r-- | pkgs/development/python-modules/pytorch/default.nix | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/pkgs/development/python-modules/pytorch/default.nix b/pkgs/development/python-modules/pytorch/default.nix index 7d56f1990a5..b0ae9ef0cf5 100644 --- a/pkgs/development/python-modules/pytorch/default.nix +++ b/pkgs/development/python-modules/pytorch/default.nix @@ -72,7 +72,7 @@ in buildPythonPackage rec { ] ++ lib.optional (pythonOlder "3.5") typing; checkPhase = '' - ${cudaStubEnv}python test/run_test.py --exclude dataloader sparse torch utils + ${cudaStubEnv}python test/run_test.py --exclude dataloader sparse torch utils distributed ''; meta = { |