python312Packages.scalene: fix build on Darwin

Co-authored-by: sarahec <sarahec@nextquestion.net>
2024-07-11 10:42:17 -07:00
parent 9e694d8b87
commit 9a517fb2f2
1 changed files with 8 additions and 2 deletions
--- a/pkgs/development/python-modules/scalene/default.nix
+++ b/pkgs/development/python-modules/scalene/default.nix
@@ -1,5 +1,6 @@
 {
  lib,
+  stdenv,
  buildPythonPackage,
  hypothesis,
  fetchPypi,
@@ -11,6 +12,7 @@
  numpy,
  psutil,
  pynvml,
+  nvidia-ml-py,
  pytestCheckHook,
  pythonOlder,
  rich,
@@ -37,9 +39,13 @@ buildPythonPackage rec {
    cloudpickle
    jinja2
    psutil
-    pynvml
    rich
-  ];
+    pynvml
+  ] ++ lib.optionals stdenv.hostPlatform.isLinux [ nvidia-ml-py ];
+
+  pythonRemoveDeps = [
+    "nvidia-ml-py3"
+  ] ++ lib.optionals stdenv.hostPlatform.isDarwin [ "nvidia-ml-py" ];

  __darwinAllowLocalNetworking = true;