about summary refs log tree commit diff
path: root/pkgs/development/python-modules/rising/default.nix
diff options
context:
space:
mode:
Diffstat (limited to 'pkgs/development/python-modules/rising/default.nix')
-rw-r--r--pkgs/development/python-modules/rising/default.nix36
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"
   ];