depot/third_party/nixpkgs/pkgs/development/python-modules/tf-keras/default.nix
Default email 159e378cbb Project import generated by Copybara.
GitOrigin-RevId: c04d5652cfa9742b1d519688f65d1bbccea9eb7e
2024-09-19 17:19:46 +03:00

53 lines
856 B
Nix

{
lib,
buildPythonPackage,
pythonOlder,
fetchPypi,
setuptools,
wheel,
numpy,
tensorflow,
pytestCheckHook,
}:
buildPythonPackage rec {
pname = "tf-keras";
version = "2.17.0";
pyproject = true;
disabled = pythonOlder "3.9";
src = fetchPypi {
pname = "tf_keras";
inherit version;
hash = "sha256-/al8GNow2g9ypafoDz7uNDsJ9MIG2tbFfJRPss0YVg4=";
};
nativeBuildInputs = [
];
pythonRelaxDeps = [
"tensorflow"
];
build-system = [
setuptools
wheel
];
dependencies = [
numpy
tensorflow
];
pythonImportsCheck = [ "tf_keras" ];
nativeCheckInputs = [ pytestCheckHook ];
meta = {
description = "Deep learning for humans";
homepage = "https://pypi.org/project/tf-keras/";
license = lib.licenses.asl20;
maintainers = with lib.maintainers; [ GaetanLepage ];
};
}