tjh.dev
1{
2 lib,
3 buildPythonPackage,
4 pythonOlder,
5 fetchPypi,
6 setuptools,
7 wheel,
8 numpy,
9 tensorflow,
10 pytestCheckHook,
11}:
12
13buildPythonPackage rec {
14 pname = "tf-keras";
15 version = "2.16.0";
16 pyproject = true;
17
18 disabled = pythonOlder "3.9";
19
20 src = fetchPypi {
21 pname = "tf_keras";
22 inherit version;
23 hash = "sha256-21OJHxrJgZfCrM7ZjNyowGuoJVZVpst+uV7UlnYRgoA=";
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}