Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

tensorflow: add tf.train.CheckpointOptions and other tf.train members. #11327

Merged
merged 22 commits into from
Feb 1, 2024
Merged
Changes from 1 commit
Commits
Show all changes
22 commits
Select commit Hold shift + click to select a range
16d48df
add CheckpointOptions
hoel-bagard Jan 27, 2024
31a3c6f
add other tf.train members
hoel-bagard Jan 27, 2024
abaad59
[pre-commit.ci] auto fixes from pre-commit.com hooks
pre-commit-ci[bot] Jan 27, 2024
2a97826
fix CI errors
hoel-bagard Jan 27, 2024
6fed7b9
fix CI errors
hoel-bagard Jan 27, 2024
bbaa547
MessageMeta -> Message
hoel-bagard Jan 27, 2024
789b115
fix CI errors
hoel-bagard Jan 27, 2024
9d381e4
Message -> MessageMeta
hoel-bagard Jan 27, 2024
56e6bcf
MessageMeta -> GeneratedProtocolMessageType
hoel-bagard Jan 27, 2024
0f7548c
Add protobuff message type (incomplete).
hoel-bagard Jan 27, 2024
450c020
[pre-commit.ci] auto fixes from pre-commit.com hooks
pre-commit-ci[bot] Jan 27, 2024
7241115
fix Checkpoint's methods
hoel-bagard Jan 27, 2024
bf6c853
comment out method and arg not present at runtime.
hoel-bagard Jan 27, 2024
66fcf63
Move messages to core/example
hoel-bagard Jan 28, 2024
8b1ba15
[pre-commit.ci] auto fixes from pre-commit.com hooks
pre-commit-ci[bot] Jan 28, 2024
6783a3e
Fix message import origin.
hoel-bagard Jan 28, 2024
8d69656
Re-add Example
hoel-bagard Jan 28, 2024
794e769
fix: disable stubtest errors for protobuf types.
hoel-bagard Jan 28, 2024
4f575fa
Merge branch 'main' into hoel/add_tf_CheckpointOptions
JelleZijlstra Jan 31, 2024
080b155
Remove ShardableTensor and ShardingCallback
hoel-bagard Jan 31, 2024
5532157
Add PythonState
hoel-bagard Jan 31, 2024
2ad4be9
remove commented out code, or add comment to explain its presence.
hoel-bagard Feb 1, 2024
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Prev Previous commit
Next Next commit
Re-add Example
hoel-bagard committed Jan 28, 2024

Verified

This commit was signed with the committer’s verified signature.
hoel-bagard Hoël Bagard
commit 8d6965637403108db6fb1885b64d90d35ebc7621
1 change: 1 addition & 0 deletions stubs/tensorflow/tensorflow/train/__init__.pyi
Original file line number Diff line number Diff line change
@@ -5,6 +5,7 @@ from typing_extensions import Self

import numpy as np
import tensorflow as tf
from tensorflow.core.example.example_pb2 import Example as Example
from tensorflow.core.example.feature_pb2 import (
BytesList as BytesList,
Feature as Feature,