From 5b6ee7bbc94a8c92c3c0079fe3b8dbe6cac3f8e2 Mon Sep 17 00:00:00 2001
From: amyeroberts <22614925+amyeroberts@users.noreply.github.com>
Date: Fri, 19 Jul 2024 15:47:20 +0100
Subject: [PATCH] Update tests/utils/test_modeling_utils.py

---
 tests/utils/test_modeling_utils.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/tests/utils/test_modeling_utils.py b/tests/utils/test_modeling_utils.py
index 71c9dde70ba71f..c47f26cffa2d83 100644
--- a/tests/utils/test_modeling_utils.py
+++ b/tests/utils/test_modeling_utils.py
@@ -815,7 +815,7 @@ def test_checkpoint_variant_local_sharded_safe(self):
         for p1, p2 in zip(model.parameters(), new_model.parameters()):
             self.assertTrue(torch.allclose(p1, p2))
 
-    def test_checkpoint_safetensors(self):
+    def test_checkpoint_loading_only_safetensors_available(self):
         # Test that the loading behaviour is as expected when only safetensor checkpoints are available
         # - We can load the model with use_safetensors=True
         # - We can load the model without specifying use_safetensors i.e. we search for the available checkpoint,