Commit Graph

72 Commits

Author SHA1 Message Date
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
Hoël Bagard
5e4483618a tensorflow: add tensorflow.bitwise (#11440)
Co-authored-by: Jelle Zijlstra <jelle.zijlstra@gmail.com>
2024-03-01 16:00:02 -08:00
Hoël Bagard
08ddbe521a tensorflow: add tensorflow.audio (#11441) 2024-03-01 07:21:10 -08:00
Hoël Bagard
ea2002d171 tensorflow: add tensorflow.keras.activations members (#11444)
Co-authored-by: Jelle Zijlstra <jelle.zijlstra@gmail.com>
2024-03-01 07:17:28 -08:00
Avasam
88238202d7 Add pyupgrade check outdated-version-block (UP036) (#11509)
Co-authored-by: Alex Waygood <Alex.Waygood@Gmail.com>
2024-03-01 07:15:53 -08:00
Hoël Bagard
791dc9120a tensorflow: add tensorflow.autodiff (#11442) 2024-02-29 06:58:47 -08:00
Hoël Bagard
d52c1f6783 tensorflow: add tensorflow.autograph (#11443) 2024-02-29 06:57:51 -08:00
Hoël Bagard
955cdf50d5 tensorflow: add tf.linalg module (#11386)
Taken from:
https://github.com/hmc-cs-mdrissi/tensorflow_stubs/blob/main/stubs/tensorflow/linalg.pyi
2024-02-17 07:45:11 -08:00
Hoël Bagard
6087745bcf tensorflow: add tf.ones, tf.zeros, tf.zeros_like and tf.ones_like functions (#11368) 2024-02-16 21:52:05 -08:00
Hoël Bagard
d93ee88ba5 tensorflow add tf.random module (#11359)
Partially from Mehdi Drissi's stubs.
2024-02-16 21:33:16 -08:00
Hoël Bagard
15fa3cf0c5 tensorflow: Add members from tensorflow.keras.metrics (#11329)
Partially taken from: https://github.com/hmc-cs-mdrissi/tensorflow_stubs/blob/main/stubs/tensorflow/keras/metrics.pyi
2024-02-16 21:13:03 -08:00
Hoël Bagard
69354d78ad tensorflow: add tf.distribute.experimental.coordinator.RemoteValue (#11349) 2024-02-16 21:11:21 -08:00
Hoël Bagard
8061e58dcf tensorflow: add tf.strings module (#11380)
Partially taken from:
https://github.com/hmc-cs-mdrissi/tensorflow_stubs/blob/main/stubs/tensorflow/strings.pyi

Co-authored-by: Jelle Zijlstra <jelle.zijlstra@gmail.com>
2024-02-16 20:08:43 -08:00
Hoël Bagard
0f0e261b90 tensorflow: Add some tf.raw_ops members. (#11387)
https://github.com/hmc-cs-mdrissi/tensorflow_stubs/blob/main/stubs/tensorflow/raw_ops.pyi
2024-02-16 19:06:56 -08:00
Hoël Bagard
55dd99a338 tensorflow Add tensorflow.summary module (#11358)
Partially derived from https://github.com/hmc-cs-mdrissi/tensorflow_stubs/blob/main/stubs/tensorflow/summary.pyi
2024-02-16 06:05:54 -08:00
Hoël Bagard
684f541357 tensorflow: Add tensorflow.keras.callbacks module (#11332) 2024-02-05 05:19:04 -08:00
Hoël Bagard
3600c805eb tensorflow fix tensorflow.data.Dataset's zip (#11357) 2024-02-04 17:23:04 -08:00
Alex Waygood
c49c84f319 A new shade of Black (#11362) 2024-02-04 16:38:55 -08:00
Hoël Bagard
7a6a749449 tensorflow: bump version to 2.15 (#11352) 2024-02-03 07:28:25 -08:00
Hoël Bagard
587e75ff88 tensorflow: add tf.train.CheckpointOptions and other tf.train members. (#11327) 2024-02-01 07:33:13 -08:00
Hoël Bagard
284afde9cc tensorflow: Add some functions from the config module (#11325) 2024-01-31 07:59:45 -08:00
Hoël Bagard
67df2df466 tensorflow: Add (and rename) aliases (#11324) 2024-01-30 20:15:15 -08:00
Hoël Bagard
98ba275d5e tensorflow: add some tensorflow functions (#11326) 2024-01-30 19:57:25 -08:00
Hoël Bagard
68ae493297 tensorflow: add tensorflow.math.reduce_variance (#11328) 2024-01-30 19:55:24 -08:00
Hoël Bagard
0f524217d6 tensorflow: fix tensorflow.VariableSynchronization (#11330) 2024-01-30 10:46:59 -08:00
Hoël Bagard
4fa759f12a Add convert_to_tensor to tensorflow (#11292) 2024-01-25 16:48:54 -08:00