pkgs/development/libraries/frugally-deep/default.nix

   1 { lib
   2 , stdenv
   3 , fetchFromGitHub
   4 , gitUpdater
   5 , cmake
   6 , functionalplus
   7 , eigen
   8 , nlohmann_json
   9 , doctest
  10 , python3Packages
  11 , buildTests ? false # Needs tensorflow
  12 }:
  13
  14 stdenv.mkDerivation (finalAttrs: {
  15   pname = "frugally-deep";
  16   version = "0.15.24-p0";
  17
  18   src = fetchFromGitHub {
  19     owner = "Dobiasd";
  20     repo = "frugally-deep";
  21     rev = "v${finalAttrs.version}";
  22     hash = "sha256-yg2SMsYOOSOgsdwIH1bU3iPM45z6c7WeIrgOddt3um4=";
  23   };
  24
  25   nativeBuildInputs = [
  26     cmake
  27   ] ++ lib.optionals buildTests [
  28     python3Packages.python
  29     python3Packages.numpy
  30   ];
  31
  32   buildInputs = lib.optionals buildTests [
  33     doctest
  34     python3Packages.tensorflow
  35   ];
  36
  37   propagatedBuildInputs = [
  38     functionalplus
  39     eigen
  40     nlohmann_json
  41   ];
  42
  43   cmakeFlags = lib.optionals buildTests [ "-DFDEEP_BUILD_UNITTEST=ON" ];
  44   passthru.updateScript = gitUpdater;
  45
  46   meta = with lib; {
  47     description = "Header-only library for using Keras (TensorFlow) models in C++";
  48     homepage = "https://github.com/Dobiasd/frugally-deep";
  49     license = with licenses; [ mit ];
  50     maintainers = with maintainers; [ Madouura ];
  51     platforms = platforms.linux;
  52   };
  53 })