Re-add Example

python · Jan 28, 2024 · 8d69656 · 8d69656
1 parent 6783a3e
commit 8d69656
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/stubs/tensorflow/tensorflow/train/__init__.pyi b/stubs/tensorflow/tensorflow/train/__init__.pyi
@@ -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,