diff --git a/datasets.py b/datasets.py index 9f23f71..b5e6731 100644 --- a/datasets.py +++ b/datasets.py @@ -9,16 +9,21 @@ def filter_reject_blacklist(mats): # matrices that are actually integer blacklist_integer = [ - "Almedar", + "Alemdar", "aug2d", "aug2dc", "aug3d", - "mc2depi" + "cyl6", + "engine", + "mc2depi", + "t520", + "tube2" ] filtered = [] for mat in mats: if mat.name in blacklist_integer: + print(f"BLACKLIST {mat.name}") continue filtered += [mat] return filtered