pkgs/development/python-modules/gpustat/default.nix

   1 {
   2   lib,
   3   blessed,
   4   buildPythonPackage,
   5   fetchPypi,
   6   mockito,
   7   nvidia-ml-py,
   8   psutil,
   9   pytestCheckHook,
  10   pythonOlder,
  11   setuptools-scm,
  12 }:
  13
  14 buildPythonPackage rec {
  15   pname = "gpustat";
  16   version = "1.1.1";
  17   format = "setuptools";
  18
  19   disabled = pythonOlder "3.7";
  20
  21   src = fetchPypi {
  22     inherit pname version;
  23     hash = "sha256-wY0+1VGPwWMAxC1pTevHCuuzvlXK6R8dtk1jtfqK+dg=";
  24   };
  25
  26   pythonRelaxDeps = [ "nvidia-ml-py" ];
  27
  28   nativeBuildInputs = [
  29     setuptools-scm
  30   ];
  31
  32   propagatedBuildInputs = [
  33     blessed
  34     nvidia-ml-py
  35     psutil
  36   ];
  37
  38   nativeCheckInputs = [
  39     mockito
  40     pytestCheckHook
  41   ];
  42
  43   pythonImportsCheck = [ "gpustat" ];
  44
  45   meta = with lib; {
  46     description = "Simple command-line utility for querying and monitoring GPU status";
  47     mainProgram = "gpustat";
  48     homepage = "https://github.com/wookayin/gpustat";
  49     changelog = "https://github.com/wookayin/gpustat/releases/tag/v${version}";
  50     license = licenses.mit;
  51     maintainers = with maintainers; [ billhuang ];
  52   };
  53 }