From 42c715a5e017e8c017199accc76d746e22e3e39d Mon Sep 17 00:00:00 2001
From: Milo Cress <iamroot@databricks.com>
Date: Tue, 17 Dec 2024 14:35:53 -0500
Subject: [PATCH] style II

---
 llmfoundry/callbacks/hf_checkpointer.py | 3 +--
 1 file changed, 1 insertion(+), 2 deletions(-)

diff --git a/llmfoundry/callbacks/hf_checkpointer.py b/llmfoundry/callbacks/hf_checkpointer.py
index 7bec404bb4..5eb6d8ca98 100644
--- a/llmfoundry/callbacks/hf_checkpointer.py
+++ b/llmfoundry/callbacks/hf_checkpointer.py
@@ -620,8 +620,7 @@ def tensor_hook(
         hooks = []
         for _, module in state_dict_model.named_modules():
             hooks.append(
-                module._register_state_dict_hook(tensor_hook),
-            )
+                module._register_state_dict_hook(tensor_hook),)
 
         state_dict = get_model_state_dict(
             state_dict_model,