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