python312Packages.torchmetrics: 1.6.1 -> 1.6.2 (#386687) (082193ea) · Commits · nix / nixpkgs

pkgs/development/python-modules/finetuning-scheduler/default.nix

+17 −5

Original line number	Diff line number	Diff line
		{
		stdenv,
		lib,
		stdenv,
		buildPythonPackage,
		fetchFromGitHub,

		# build-system
		setuptools,

		# dependencies
		pytorch-lightning,
		torch,

		# tests
		pythonOlder,
		pythonAtLeast,
		pytestCheckHook,
		torch,
		pytorch-lightning,
		}:

		buildPythonPackage rec {
		@@ -39,13 +46,18 @@ buildPythonPackage rec {
		nativeCheckInputs = [ pytestCheckHook ];
		pytestFlagsArray = [ "tests" ];
		disabledTests =
		lib.optionals (pythonOlder "3.12") [
		# torch._dynamo.exc.BackendCompilerFailed: backend='inductor' raised:
		# LoweringException: ImportError: cannot import name 'triton_key' from 'triton.compiler.compiler'
		lib.optionals (pythonOlder "3.12") [
		"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"

+2 −2

Original line number	Diff line number	Diff line
		@@ -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 = [