try to fix import

python · Feb 4, 2024 · b07ed76 · b07ed76
1 parent e51face
commit b07ed76
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 0 deletions.
diff --git a/stubs/tensorflow/tensorflow/__init__.pyi b/stubs/tensorflow/tensorflow/__init__.pyi
@@ -11,6 +11,7 @@ from typing_extensions import ParamSpec, Self
 from google.protobuf.message import Message
 from tensorflow import (
     data as data,
+    distribute as distribute,
     experimental as experimental,
     feature_column as feature_column,
     initializers as initializers,

diff --git a/stubs/tensorflow/tensorflow/distribute/__init__.pyi b/stubs/tensorflow/tensorflow/distribute/__init__.pyi
@@ -0,0 +1 @@
+from tensorflow.distribute import experimental as experimental
Original file line number	Diff line number	Diff line change
		@@ -0,0 +1 @@
		from tensorflow.distribute import experimental as experimental