From 0f524217d62e43d751c189d04ab8d3623a60f992 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Ho=C3=ABl=20Bagard?=
 <34478245+hoel-bagard@users.noreply.github.com>
Date: Wed, 31 Jan 2024 03:46:59 +0900
Subject: [PATCH] `tensorflow`: fix `tensorflow.VariableSynchronization`
 (#11330)

---
 stubs/tensorflow/tensorflow/__init__.pyi | 8 ++++----
 1 file changed, 4 insertions(+), 4 deletions(-)

diff --git a/stubs/tensorflow/tensorflow/__init__.pyi b/stubs/tensorflow/tensorflow/__init__.pyi
index 18f49fe37..9efff3cf3 100644
--- a/stubs/tensorflow/tensorflow/__init__.pyi
+++ b/stubs/tensorflow/tensorflow/__init__.pyi
@@ -137,10 +137,10 @@ class VariableSynchronization(Enum):
     ON_READ = 3
 
 class VariableAggregation(Enum):
-    AUTO = 0
-    NONE = 1
-    ON_WRITE = 2
-    ON_READ = 3
+    NONE = 0
+    SUM = 1
+    MEAN = 2
+    ONLY_FIRST_REPLICA = 3
 
 class _VariableMetaclass(type): ...