1{
2 lib,
3 buildPythonPackage,
4 fetchPypi,
5 poetry-core,
6 numpy,
7 scipy,
8 torch,
9}:
10
11buildPythonPackage 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}