def setup_context(self, query, context):
    for ent in self._entities:
        ent.setup_context(query, context)
def setup_context(self, query, context):
    for ent in self._entities:
        ent.setup_context(query, context)