Only look for classes when we load them
currently we get this error
elem_classes = [c for c in elt["classes"] if c["state"] != "rejected"]
TypeError: 'NoneType' object is not iterable
when running the data generator without specifying any accepted classes