Make check a little more stringent

huggingface · Feb 28, 2025 · 769cdc3 · 769cdc3
1 parent b6ad9d1
commit 769cdc3
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/peft_integration/test_peft_integration.py b/tests/peft_integration/test_peft_integration.py
@@ -531,7 +531,7 @@ def test_peft_pipeline(self):
             base_pipe = pipeline("text-generation", base_model_id)
             peft_params = list(peft_pipe.model.parameters())
             base_params = list(base_pipe.model.parameters())
-            self.assertNotEqual(peft_params, base_params)  # Assert we actually loaded the adapter too
+            self.assertNotEqual(len(peft_params), len(base_params))  # Assert we actually loaded the adapter too
             _ = peft_pipe("Hello")
 
     def test_peft_add_adapter_with_state_dict(self):