tjh.dev
1{ buildPythonPackage, fetchPypi, lib, numpy, pytest, pytestpep8, pytestcov }:
2buildPythonPackage rec {
3 version = "2.3.2";
4 pname = "opt_einsum";
5
6 src = fetchPypi {
7 inherit version pname;
8 sha256 = "0ny3v8x83mzpwmqjdzqhzy2pzwyy4wx01r1h9i29xw3yvas69m6k";
9 };
10
11 checkInputs = [
12 pytest
13 pytestpep8
14 pytestcov
15 ];
16
17 checkPhase = ''
18 pytest
19 '';
20
21 propagatedBuildInputs = [
22 numpy
23 ];
24
25 meta = {
26 description = "Optimizing NumPy's einsum function with order optimization and GPU support.";
27 homepage = http://optimized-einsum.readthedocs.io;
28 license = lib.licenses.mit;
29 maintainers = with lib.maintainers; [ teh ];
30 };
31}