python312Packages.torchmetrics: 1.6.1 -> 1.6.2 (#386687)

This commit is contained in:
Yt 2025-03-03 17:45:07 -05:00 committed by GitHub
commit 082193eab1
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
2 changed files with 20 additions and 8 deletions

View file

@ -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"

View file

@ -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 = [