Unverified Commit 8cdf7e75 authored by Someone's avatar Someone Committed by GitHub
Browse files

Merge pull request #329511 from trofi/python3Packages.pytorch-bin.tests-fix-eval 

python3Packages.pytorch-bin.tests: fix the eval
parents af674633 f84f1c27

pkgs/development/python-modules/torch/bin.nix

+1 −4
Original line number Diff line number Diff line
@@ -121,10 +121,7 @@ buildPythonPackage { 


  pythonImportsCheck = [ "torch" ];



  passthru.tests = callPackage ./tests.nix {

    torchWithCuda = torch-bin;

    torchWithRocm = torch-bin;

  };

  passthru.tests = callPackage ./tests.nix {};



  meta = {

    description = "PyTorch: Tensors and Dynamic neural networks in Python with strong GPU acceleration";