From 5470e5f94e302b1a60ee2c1add0caf5f0e879e42 Mon Sep 17 00:00:00 2001
From: genghaozhe <939857490@qq.com>
Date: Thu, 16 May 2024 08:03:40 +0000
Subject: [PATCH] a commit for fake push test

---
 tests/test_zero/test_gemini/test_optim.py | 4 +---
 1 file changed, 1 insertion(+), 3 deletions(-)

diff --git a/tests/test_zero/test_gemini/test_optim.py b/tests/test_zero/test_gemini/test_optim.py
index a9366e7bc..1c914ca0e 100644
--- a/tests/test_zero/test_gemini/test_optim.py
+++ b/tests/test_zero/test_gemini/test_optim.py
@@ -40,9 +40,7 @@ EXAMPLE_MODELS = [
 ]
 
 # bfloat16 cannot represent them exactly
-BF16_IGNORED_KEYS = [
-    "masked_bias",
-]
+BF16_IGNORED_KEYS = ["masked_bias"]
 
 
 def check_param(model: GeminiDDP, torch_model: torch.nn.Module, dtype: torch.dtype):