pkgs/development/python-modules/tf-keras/default.nix

   1 {
   2   lib,
   3   buildPythonPackage,
   4   pythonOlder,
   5   fetchPypi,
   6   setuptools,
   7   wheel,
   8   numpy,
   9   tensorflow,
  10   pytestCheckHook,
  11 }:
  12
  13 buildPythonPackage rec {
  14   pname = "tf-keras";
  15   version = "2.17.0";
  16   pyproject = true;
  17
  18   disabled = pythonOlder "3.9";
  19
  20   src = fetchPypi {
  21     pname = "tf_keras";
  22     inherit version;
  23     hash = "sha256-/al8GNow2g9ypafoDz7uNDsJ9MIG2tbFfJRPss0YVg4=";
  24   };
  25
  26   nativeBuildInputs = [
  27   ];
  28
  29   pythonRelaxDeps = [
  30     "tensorflow"
  31   ];
  32
  33   build-system = [
  34     setuptools
  35     wheel
  36   ];
  37
  38   dependencies = [
  39     numpy
  40     tensorflow
  41   ];
  42
  43   pythonImportsCheck = [ "tf_keras" ];
  44
  45   nativeCheckInputs = [ pytestCheckHook ];
  46
  47   meta = {
  48     description = "Deep learning for humans";
  49     homepage = "https://pypi.org/project/tf-keras/";
  50     license = lib.licenses.asl20;
  51     maintainers = with lib.maintainers; [ GaetanLepage ];
  52   };
  53 }