diff --git a/pytools/__init__.py b/pytools/__init__.py index f480c2cfbc3f7d76ff4451a53e7947b025945783..17550f668cba40a26aafeb908925d22b248fb44e 100644 --- a/pytools/__init__.py +++ b/pytools/__init__.py @@ -157,6 +157,14 @@ class RecordWithoutPickling(object): for fld in self.__class__.fields if hasattr(self, fld))) + def register_fields(self, new_fields): + try: + fields = self.__class__.fields + except AttributeError: + self.__class__.fields = fields = set() + + fields.update(new_fields) + class Record(RecordWithoutPickling): __slots__ = []