diff options
Diffstat (limited to 'pkgs/development/python-modules/rising/default.nix')
-rw-r--r-- | pkgs/development/python-modules/rising/default.nix | 36 |
1 files changed, 24 insertions, 12 deletions
diff --git a/pkgs/development/python-modules/rising/default.nix b/pkgs/development/python-modules/rising/default.nix index 85990fda48098..49b38e2e3426b 100644 --- a/pkgs/development/python-modules/rising/default.nix +++ b/pkgs/development/python-modules/rising/default.nix @@ -1,48 +1,60 @@ { - stdenv, lib, buildPythonPackage, - pythonOlder, fetchFromGitHub, - pytestCheckHook, - pythonRelaxDepsHook, - dill, + + # build-system + versioneer, + + # dependencies lightning-utilities, numpy, torch, threadpoolctl, tqdm, + + # tests + dill, + pytestCheckHook, + + stdenv, }: 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="; }; - nativeBuildInputs = [ pythonRelaxDepsHook ]; - pythonRelaxDeps = [ "lightning-utilities" ]; - propagatedBuildInputs = [ + # Remove vendorized versioneer (incompatible with python 3.12) + postPatch = '' + rm versioneer.py + ''; + + build-system = [ versioneer ]; + + dependencies = [ lightning-utilities numpy torch threadpoolctl tqdm ]; + nativeCheckInputs = [ dill pytestCheckHook ]; - disabledTests = lib.optionals (stdenv.isLinux && stdenv.isAarch64) [ + + disabledTests = lib.optionals (stdenv.hostPlatform.isLinux && stdenv.hostPlatform.isAarch64) [ # RuntimeError: DataLoader worker (pid(s) <...>) exited unexpectedly: "test_progressive_resize_integration" ]; |