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

   1 {
   2   lib,
   3   buildPythonPackage,
   4   fetchPypi,
   5   poetry-core,
   6   numpy,
   7   scipy,
   8   torch,
   9 }:
  10
  11 buildPythonPackage rec {
  12   pname = "dctorch";
  13   version = "0.1.2";
  14   pyproject = true;
  15
  16   src = fetchPypi {
  17     inherit pname version;
  18     hash = "sha256-TmfLAkiofrQNWYBhIlY4zafbZPgFftISCGloO/rlEG4=";
  19   };
  20
  21   nativeBuildInputs = [ poetry-core ];
  22
  23   propagatedBuildInputs = [
  24     numpy
  25     scipy
  26     torch
  27   ];
  28
  29   pythonImportsCheck = [ "dctorch" ];
  30
  31   doCheck = false; # no tests
  32
  33   meta = with lib; {
  34     description = "Fast discrete cosine transforms for pytorch";
  35     homepage = "https://pypi.org/project/dctorch/";
  36     license = licenses.mit;
  37     maintainers = teams.tts.members;
  38   };
  39 }