Compare commits

...

2 Commits

Author SHA1 Message Date
e68eefcc2f fix 2023-11-09 11:19:13 +01:00
f84c122c04 fix 2023-11-09 10:27:04 +01:00
2 changed files with 3 additions and 1 deletions

View File

@ -3,7 +3,7 @@ name: Check Tiny Models
on:
push:
branches:
- check_tiny_models*
- run_tiny_with_fix_tiny_model_creation
repository_dispatch:
schedule:
- cron: "0 2 * * *"

View File

@ -81,6 +81,8 @@ def get_test_classes(test_file):
test_module = get_test_module(test_file)
for attr in dir(test_module):
attr_value = getattr(test_module, attr)
if not (isinstance(attr_value, type) and "ModelTesterMixin" in [x.__name__ for x in attr_value.__bases__]):
continue
# (TF/Flax)ModelTesterMixin is also an attribute in specific model test module. Let's exclude them by checking
# `all_model_classes` is not empty (which also excludes other special classes).
model_classes = getattr(attr_value, "all_model_classes", [])