pkgs/development/python-modules/nianet/default.nix

   1 { lib
   2 , buildPythonPackage
   3 , fetchFromGitHub
   4 , matplotlib
   5 , niapy
   6 , numpy
   7 , poetry-core
   8 , pytestCheckHook
   9 , pythonOlder
  10 , scikit-learn
  11 , toml-adapt
  12 , tomli
  13 , torch
  14 }:
  15
  16 buildPythonPackage rec {
  17   pname = "nianet";
  18   version = "1.1.4";
  19   pyproject = true;
  20
  21   disabled = pythonOlder "3.9";
  22
  23   src = fetchFromGitHub {
  24     owner = "SasoPavlic";
  25     repo = "nianet";
  26     rev = "refs/tags/version_${version}";
  27     sha256 = "sha256-FZipl6Z9AfiL6WH0kvUn8bVxt8JLdDVlmTSqnyxe0nY=";
  28   };
  29
  30   nativeBuildInputs = [
  31     poetry-core
  32     toml-adapt
  33   ];
  34
  35   propagatedBuildInputs = [
  36     niapy
  37     numpy
  38     scikit-learn
  39     torch
  40   ];
  41
  42   # create niapy and torch dep version consistent
  43   preBuild = ''
  44     toml-adapt -path pyproject.toml -a change -dep niapy -ver X
  45     toml-adapt -path pyproject.toml -a change -dep torch -ver X
  46   '';
  47
  48   nativeCheckInputs = [
  49     pytestCheckHook
  50     tomli
  51   ];
  52
  53   pythonImportsCheck = [
  54     "nianet"
  55   ];
  56
  57   meta = with lib; {
  58     description = "Designing and constructing neural network topologies using nature-inspired algorithms";
  59     homepage = "https://github.com/SasoPavlic/NiaNet";
  60     changelog = "https://github.com/SasoPavlic/NiaNet/releases/tag/v${version}";
  61     license = licenses.mit;
  62     maintainers = with maintainers; [ firefly-cpp ];
  63   };
  64 }