tjh.dev
1{
2 lib,
3 buildPythonPackage,
4 pythonOlder,
5 fetchFromGitHub,
6 torch,
7}:
8
9buildPythonPackage rec {
10 pname = "lion-pytorch";
11 version = "0.2.2";
12 format = "setuptools";
13 disabled = pythonOlder "3.6";
14
15 src = fetchFromGitHub {
16 owner = "lucidrains";
17 repo = "lion-pytorch";
18 rev = "refs/tags/${version}";
19 hash = "sha256-hOPTuXdTrTi/Thv3/5IYqkgH+5cFdzyK1Fshus8u5n0=";
20 };
21
22 propagatedBuildInputs = [ torch ];
23
24 pythonImportsCheck = [ "lion_pytorch" ];
25 doCheck = false; # no tests currently
26
27 meta = with lib; {
28 description = "Optimizer tuned by Google Brain using genetic algorithms";
29 homepage = "https://github.com/lucidrains/lion-pytorch";
30 license = licenses.mit;
31 maintainers = with maintainers; [ bcdarwin ];
32 };
33}