Loading pkgs/development/python-modules/finetuning-scheduler/default.nix +12 −4 Original line number Diff line number Diff line Loading @@ -36,7 +36,15 @@ buildPythonPackage rec { nativeCheckInputs = [ pytestCheckHook ]; pytestFlagsArray = [ "tests" ]; disabledTests = lib.optionals (stdenv.isAarch64 && stdenv.isLinux) [ 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" ]; Loading pkgs/development/python-modules/lightning-utilities/default.nix +2 −4 Original line number Diff line number Diff line Loading @@ -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 = '' Loading pkgs/development/python-modules/rising/default.nix +17 −9 Original line number Diff line number Diff line { 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" Loading @@ -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"; }; } Loading
pkgs/development/python-modules/finetuning-scheduler/default.nix +12 −4 Original line number Diff line number Diff line Loading @@ -36,7 +36,15 @@ buildPythonPackage rec { nativeCheckInputs = [ pytestCheckHook ]; pytestFlagsArray = [ "tests" ]; disabledTests = lib.optionals (stdenv.isAarch64 && stdenv.isLinux) [ 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" ]; Loading
pkgs/development/python-modules/lightning-utilities/default.nix +2 −4 Original line number Diff line number Diff line Loading @@ -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 = '' Loading
pkgs/development/python-modules/rising/default.nix +17 −9 Original line number Diff line number Diff line { 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" Loading @@ -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"; }; }