diff options
Diffstat (limited to 'pkgs/development/python-modules/scalene/default.nix')
-rw-r--r-- | pkgs/development/python-modules/scalene/default.nix | 14 |
1 files changed, 10 insertions, 4 deletions
diff --git a/pkgs/development/python-modules/scalene/default.nix b/pkgs/development/python-modules/scalene/default.nix index 1d635edbedd9d..6f5adae67d84f 100644 --- 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, @@ -18,13 +20,13 @@ buildPythonPackage rec { pname = "scalene"; - version = "1.5.41"; + version = "1.5.43.2"; pyproject = true; disabled = pythonOlder "3.9"; src = fetchPypi { inherit pname version; - hash = "sha256-akjxv9Qot2lGntZxkxfFqz65VboL1qduykfjyEg1Ivg="; + hash = "sha256-LtD7v3pLz4UCnh6xlhkPdcEjyu3mt+YQPYZ0nNCLuDw="; }; nativeBuildInputs = [ @@ -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; |