From 0779fc8eb8e7c86d11c2fab5637d06d9ec907daf Mon Sep 17 00:00:00 2001 From: Connor Henderson Date: Tue, 25 Jul 2023 12:24:36 -0400 Subject: [PATCH] Edit err message and comment in `test_model_is_small` (#25087) * Edit err message and comment in * put back 80M comment --- tests/test_modeling_common.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/test_modeling_common.py b/tests/test_modeling_common.py index 87960983f9..d80624b718 100755 --- 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()