def remove(self, entity):
        # TODO this doesn't take direction into account
        # TODO also, super clunky
        for rel in list(self.relation_set):
            if rel.from_entity in entity:
                rel.detach()
dataset/ETHPy150Open eevee/flax/flax/relation.py/RelationProxy.remove/operator `is` -> `in`