Commit Graph

97 Commits

Author SHA1 Message Date
lev-blit fc7a5448ff Rename requires to dependencies in METADATA files (#15594) 2026-04-01 19:15:12 +02:00
Semyon Moroz 282b1a838a Add @type_check_only to stub-only private classes in various third-party stubs (#15535) 2026-03-21 18:02:17 -04:00
Semyon Moroz 33b3568a93 [tensorflow] Add __slots__ (#15459) 2026-02-23 20:22:06 +01:00
Hoël Bagard a3568ac8d9 [tensorflow] Fix tf.tile (#15306) 2026-01-20 08:59:09 +01:00
Hoël Bagard 2bcf7bb270 [tensorflow]: Add tf.tile (#15300) 2026-01-19 15:55:16 +01:00
Hoël Bagard 64f965e5aa tensorflow: add Activation, GlobalAveragePooling2D and MaxPool2D layers (#15302) 2026-01-19 09:23:31 +01:00
Hoël Bagard 3789c36681 tensorflow: add tf.keras.metrics.MeanSquaredError (#15282) 2026-01-15 08:34:40 +01:00
Hoël Bagard 47c0e2d5aa [tensorflow]: Add a few missing elements (#15265) 2026-01-12 15:32:25 +01:00
Hoël Bagard 31791106cb [tensorflow]: Add tf.split (#14890) 2026-01-12 14:45:31 +01:00
Hoël Bagard 7b277d8516 tensorflow: add missing re-export (#14816) 2025-10-07 12:31:46 +02:00
Sebastian Rittau 6a7f3190b1 Remove @override annotations (#14806) 2025-09-30 13:47:57 +02:00
EthanAtLL ba2e6b454a tensorflow: Add Tensor.__array__ method. (#14342) 2025-08-08 18:29:30 +02:00
Sebastian Rittau 981bd8eb8d Remove pytype workarounds from third-party stubs (#14471) 2025-07-27 13:05:57 +02:00
Sebastian Rittau 126768408a Replace Incomplete | None = None in third party stubs (#14063) 2025-05-15 21:37:43 +02:00
Sebastian Rittau e08e349699 Replace incomplete module markers (#14030) 2025-05-13 08:53:39 +02:00
Avasam 5c7fe07278 Run Black on *_pb2 files (#13930) 2025-05-05 15:22:59 +02:00
Avasam f26b41bdd1 Mention rerunning sync_protobuf scripts (#13924) 2025-05-03 02:29:31 +02:00
Julen Costa Watanabe 23c2cba7ff [tensorflow] Fix literal value for pad function (#13877) 2025-04-24 13:56:57 +02:00
Avasam 4b253d2932 Enable Ruff flake8-todos (TD) (#13748) 2025-04-03 07:43:54 +02:00
Sebastian Rittau 414e6d2150 [CI] Run stubtest with Python 3.13 (#13638) 2025-03-18 13:36:15 +01:00
Avasam 6d6e858e63 Enable Ruff PLC (Pylint Convention) (#13306) 2025-03-03 15:39:40 +01:00
Hoël Bagard 20aef1d78f tensorflow: Change Tensor.__bool__ return from NoReturn to bool (#13562) 2025-02-28 19:20:14 +01:00
Hoël Bagard 8c6493925f tensorflow: Allow tf.Tensor to be used in __getitem__ (#13549) 2025-02-27 12:19:20 +01:00
Hoël Bagard a1c185b0b2 tensorflow: add a few TensorFlow functions (#13364) 2025-02-26 15:07:07 +01:00
Avasam dde13fb91d Enable Ruff PGH rules (#13304) 2024-12-26 21:35:34 +00:00
Alex Waygood 90e6bd9aca Pin keras to 3.6.* when stubtesting tensorflow (#13137) 2024-11-27 08:50:23 -08:00
Ali Hamdan a870a4e63a Remove numpy pins from several stubs (#12993) 2024-11-11 17:47:11 +01:00
Avasam 335cc91b70 Bump tensorflow to ~=2.18.0 (#12916)
* Tensorflow proto script update

* Manual stubtest changes

* Use Path for arg type
2024-10-28 20:32:31 -07:00
Avasam bff5547abc Use lib/ts_utils for scripts/sync_protobuf (#12913)
- Simplified `download_file` error handling (400+ return codes already raised errors!)
- Moved `update_metadata` from `scripts/sync_protobuf/_utils.py` to `lib/ts_utils/metadata.py`
- Improved `update_metadata` to support any key (values unvalidated atm) and return the modified dictionary
- Updated `scripts/stubsabot.py` to use `update_metadata`
- Updated `scripts/sync_protobuf/*` to use  `lib/ts_utils`
- Updated `scripts/sync_protobuf/tensorflow.py` and `scripts/sync_protobuf/google_protobuf.py` to use the version directly from the `METADATA.toml` file
2024-10-28 11:47:52 +01:00
Avasam c025e37bbb Rewrote protobuf generation scripts in Python (#12527) 2024-09-19 08:11:21 +02:00
sobolevn 30a20398c9 Pin numpy in various stubs (#12554) 2024-08-19 15:38:23 +03:00
Max Muoto 37807d753a Use Generator instead of Iterator for 3rd-party context managers (#12481) 2024-08-12 14:26:18 +02:00
Avasam d3ca513ddb Bump tensorflow to 2.17.* (#12512) 2024-08-12 13:39:34 +02:00
Ali Hamdan 0bc7b4fd5a Remove numpy version pin for several stubs(#12476) 2024-08-03 11:52:55 +01:00
Sebastian Rittau fa73f9ff4a [tensorflow] Pin keras to 3.3.* (#12221) 2024-06-26 11:03:32 +02:00
Sebastian Rittau 212af7a526 Pin various stubs to numpy to < 2 (#12152)
Fixes #12146
2024-06-17 15:50:22 +02:00
Ali Hamdan 77ef4d639e Replace np.float_ alias removed in numpy 2.0 (#12138) 2024-06-16 11:45:25 +02:00
Sam fd67f0a457 Add stubs for tf.math.angle to math.pyi (#12099) 2024-06-05 23:06:54 +02:00
Alex Waygood 2db0b6e655 Fix stubtest for TensorFlow with latest keras release (#11838) 2024-04-27 09:42:34 +01:00
Alex Waygood c13f6e166d Fix stubtest errors in tensorflow with keras>=3.3.2 (#11817) 2024-04-23 13:26:02 +02:00
Avasam b3bfdadb45 Bump tensorflow to 2.16.* (#11696) 2024-04-22 16:37:33 +02:00
Avasam 1017c525f8 Simplify protoc install in protobuf generation scripts (#11785) 2024-04-21 19:33:31 +02:00
Sebastian Rittau 7d56cd9a6c Remove remaining bare Incompletes (#11768)
Enable Y065
2024-04-16 06:26:14 -07:00
Alex Waygood 6883b80f52 Bump flake8-pyi to 24.4.0 (#11745) 2024-04-11 13:05:46 +01:00
Avasam fe02cba606 Bump mypy-protobuf in sync_tensorflow script and improve generation scripts (#11740) 2024-04-10 09:12:07 -04:00
Hoël Bagard 8558307188 tensorflow: Add tensorflow.keras.models.Model (#11334)
Based on:

- https://github.com/hmc-cs-mdrissi/tensorflow_stubs/blob/main/stubs/tensorflow/saved_model/__init__.pyi
- https://github.com/hmc-cs-mdrissi/tensorflow_stubs/blob/main/stubs/tensorflow/types/experimental.pyi
2024-03-12 21:01:09 -07:00
Hoël Bagard 6c52322be6 tensorflow: Add missing members to the tensorflow.keras.layers module. (#11333)
Co-authored-by: Jelle Zijlstra <jelle.zijlstra@gmail.com>
Co-authored-by: Rebecca Chen <rechen@google.com>
2024-03-12 20:45:47 -07:00
Hoël Bagard 56bb53aea7 tensorflow add tensorflow.saved_model (#11439)
Based on:
- https://github.com/hmc-cs-mdrissi/tensorflow_stubs/blob/main/stubs/tensorflow/saved_model/__init__.pyi
- https://github.com/hmc-cs-mdrissi/tensorflow_stubs/blob/main/stubs/tensorflow/types/experimental.pyi
2024-03-12 20:30:11 -07:00
Shantanu 88fa182253 Use PEP 570 syntax in third party stubs (#11554) 2024-03-10 14:11:43 +01:00
Hoël Bagard 176c89c06f tensorflow: add partial tf.nn module. (#11388)
Some derived from:
https://github.com/hmc-cs-mdrissi/tensorflow_stubs/blob/main/stubs/tensorflow/summary.pyi
https://github.com/hmc-cs-mdrissi/tensorflow_stubs/blob/main/stubs/tensorflow/nn.pyi
2024-03-01 23:22:43 -08:00