Edit err message and comment in `test_model_is_small` (#25087)

* Edit err message and comment in * put back 80M comment

Edit err message and comment in `test_model_is_small` (#25087)
* Edit err message and comment in * put back 80M comment
0779fc8e · Connor Henderson · GitHub · 2fac3422 · 0779fc8e
Unverified Commit 0779fc8e authored Jul 25, 2023 by Connor Henderson Committed by GitHub Jul 25, 2023
Show whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

tests/test_modeling_common.py tests/test_modeling_common.py +1 -1

No files found.
--- a/tests/test_modeling_common.py
+++ b/tests/test_modeling_common.py
@@ -2714,7 +2714,7 @@ class ModelTesterMixin:
            num_params = model.num_parameters()
            assert (
                num_params < 1000000
-            ), f"{model_class} is too big for the common tests ({num_params})! It should have 200k max."
+            ), f"{model_class} is too big for the common tests ({num_params})! It should have 1M max."


 global_rng = random.Random()