Unverified Commit 071f6a5f authored by Madoura's avatar Madoura
Browse files

python3Packages.pytorch-pfn-extras: fixup tests for python 3.11 

Revert this commit when https://github.com/NixOS/nixpkgs/pull/259068 is merged
parent a92172bf

pkgs/development/python-modules/pytorch-pfn-extras/default.nix

+4 −0
Original line number Diff line number Diff line
@@ -8,6 +8,7 @@ 
, torch

, torchvision

, typing-extensions

, pythonAtLeast

}:



buildPythonPackage rec {
@@ -60,6 +61,9 @@ buildPythonPackage rec { 
    "tests/pytorch_pfn_extras_tests/utils_tests/test_checkpoint.py"

    "tests/pytorch_pfn_extras_tests/utils_tests/test_comparer.py"

    "tests/pytorch_pfn_extras_tests/utils_tests/test_new_comparer.py"

  ] ++ lib.optionals (pythonAtLeast "3.11") [

    # Remove this when https://github.com/NixOS/nixpkgs/pull/259068 is merged

    "tests/pytorch_pfn_extras_tests/dynamo_tests/test_compile.py"

  ];



  meta = with lib; {