python312Packages.lightning-utilities: 0.11.6 -> 0.11.7 (#339111)

This commit is contained in:
Adam C. Stephens
2024-09-04 20:01:46 -04:00
committed by GitHub
3 changed files with 31 additions and 17 deletions
@@ -36,10 +36,18 @@ buildPythonPackage rec {
nativeCheckInputs = [ pytestCheckHook ];
pytestFlagsArray = [ "tests" ];
disabledTests = lib.optionals (stdenv.isAarch64 && stdenv.isLinux) [
# slightly exceeds numerical tolerance on aarch64-linux:
"test_fts_frozen_bn_track_running_stats"
];
disabledTests =
# 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 (stdenv.isAarch64 && stdenv.isLinux) [
# slightly exceeds numerical tolerance on aarch64-linux:
"test_fts_frozen_bn_track_running_stats"
];
pythonImportsCheck = [ "finetuning_scheduler" ];
@@ -19,16 +19,14 @@
buildPythonPackage rec {
pname = "lightning-utilities";
version = "0.11.6";
version = "0.11.7";
pyproject = true;
disabled = pythonOlder "3.8";
src = fetchFromGitHub {
owner = "Lightning-AI";
repo = "utilities";
rev = "refs/tags/v${version}";
hash = "sha256-Bw28mLz9GaMucqP+EqR1F3OKLxDJiIVacrCBIV35G/I=";
hash = "sha256-0XxBDe9OGQLfl4viuUm5Hx8WvZhSj+J0FoDqD/JOiZM=";
};
postPatch = ''
@@ -1,45 +1,51 @@
{
stdenv,
lib,
buildPythonPackage,
pythonOlder,
fetchFromGitHub,
pytestCheckHook,
dill,
# dependencies
lightning-utilities,
numpy,
torch,
threadpoolctl,
tqdm,
# tests
dill,
pytestCheckHook,
stdenv,
pythonAtLeast,
}:
buildPythonPackage rec {
pname = "rising";
version = "0.3.0";
format = "setuptools";
disabled = pythonOlder "3.8";
pyproject = true;
src = fetchFromGitHub {
owner = "PhoenixDL";
repo = pname;
repo = "rising";
rev = "refs/tags/v${version}";
hash = "sha256-sBzVTst5Tp2oZZ+Xsg3M7uAMbucL6idlpYwHvib3EaY=";
};
pythonRelaxDeps = [ "lightning-utilities" ];
propagatedBuildInputs = [
dependencies = [
lightning-utilities
numpy
torch
threadpoolctl
tqdm
];
nativeCheckInputs = [
dill
pytestCheckHook
];
disabledTests = lib.optionals (stdenv.isLinux && stdenv.isAarch64) [
# RuntimeError: DataLoader worker (pid(s) <...>) exited unexpectedly:
"test_progressive_resize_integration"
@@ -60,5 +66,7 @@ buildPythonPackage rec {
homepage = "https://rising.rtfd.io";
license = lib.licenses.mit;
maintainers = with lib.maintainers; [ bcdarwin ];
# AttributeError: module 'configparser' has no attribute 'SafeConfigParser'. Did you mean: 'RawConfigParser'?
broken = pythonAtLeast "3.12";
};
}