pyrox.dev
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 pythonRelaxDeps = [ "numpy" ];
22
23 build-system = [ poetry-core ];
24
25 dependencies = [
26 numpy
27 scipy
28 torch
29 ];
30
31 pythonImportsCheck = [ "dctorch" ];
32
33 doCheck = false; # no tests
34
35 meta = with lib; {
36 description = "Fast discrete cosine transforms for pytorch";
37 homepage = "https://pypi.org/project/dctorch/";
38 license = licenses.mit;
39 teams = [ teams.tts ];
40 };
41}