pkgs/development/python-modules/lion-pytorch/default.nix

   1 {
   2   lib,
   3   buildPythonPackage,
   4   pythonOlder,
   5   fetchFromGitHub,
   6   torch,
   7 }:
   8
   9 buildPythonPackage 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 }