mirror of
https://github.com/NixOS/nixpkgs.git
synced 2025-06-10 19:55:41 +03:00
python312Packages.torchmetrics: 1.6.1 -> 1.6.2 (#386687)
This commit is contained in:
commit
082193eab1
2 changed files with 20 additions and 8 deletions
|
@ -1,13 +1,20 @@
|
|||
{
|
||||
stdenv,
|
||||
lib,
|
||||
stdenv,
|
||||
buildPythonPackage,
|
||||
fetchFromGitHub,
|
||||
|
||||
# build-system
|
||||
setuptools,
|
||||
pythonOlder,
|
||||
pytestCheckHook,
|
||||
torch,
|
||||
|
||||
# dependencies
|
||||
pytorch-lightning,
|
||||
torch,
|
||||
|
||||
# tests
|
||||
pythonOlder,
|
||||
pythonAtLeast,
|
||||
pytestCheckHook,
|
||||
}:
|
||||
|
||||
buildPythonPackage rec {
|
||||
|
@ -39,13 +46,18 @@ buildPythonPackage rec {
|
|||
nativeCheckInputs = [ pytestCheckHook ];
|
||||
pytestFlagsArray = [ "tests" ];
|
||||
disabledTests =
|
||||
# torch._dynamo.exc.BackendCompilerFailed: backend='inductor' raised:
|
||||
# LoweringException: ImportError: cannot import name 'triton_key' from 'triton.compiler.compiler'
|
||||
lib.optionals (pythonOlder "3.12") [
|
||||
# torch._dynamo.exc.BackendCompilerFailed: backend='inductor' raised:
|
||||
# LoweringException: ImportError: cannot import name 'triton_key' from 'triton.compiler.compiler'
|
||||
"test_fts_dynamo_enforce_p0"
|
||||
"test_fts_dynamo_resume"
|
||||
"test_fts_dynamo_intrafit"
|
||||
]
|
||||
++ lib.optionals (pythonAtLeast "3.13") [
|
||||
# RuntimeError: Dynamo is not supported on Python 3.13+
|
||||
"test_fts_dynamo_enforce_p0"
|
||||
"test_fts_dynamo_resume"
|
||||
]
|
||||
++ lib.optionals (stdenv.hostPlatform.isAarch64 && stdenv.hostPlatform.isLinux) [
|
||||
# slightly exceeds numerical tolerance on aarch64-linux:
|
||||
"test_fts_frozen_bn_track_running_stats"
|
||||
|
|
|
@ -24,14 +24,14 @@
|
|||
|
||||
buildPythonPackage rec {
|
||||
pname = "torchmetrics";
|
||||
version = "1.6.1";
|
||||
version = "1.6.2";
|
||||
pyproject = true;
|
||||
|
||||
src = fetchFromGitHub {
|
||||
owner = "Lightning-AI";
|
||||
repo = "torchmetrics";
|
||||
tag = "v${version}";
|
||||
hash = "sha256-itLFJB920hQGX2VLOLolHhmXFVHAOkfRRFtUGB9neKM=";
|
||||
hash = "sha256-IPhBgGciB/3IFYVX+miID4oQ+5CpcjvAHiufxKiHzFw=";
|
||||
};
|
||||
|
||||
dependencies = [
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue