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

Revert this commit when https://github.com/NixOS/nixpkgs/pull/259068 is merged
This commit is contained in:
Madoura 2023-10-24 13:08:28 -05:00
parent a92172bf65
commit 071f6a5fb2
No known key found for this signature in database
GPG Key ID: 1B9BB76A2B71922D

View File

@ -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; {