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