Method __getattr__

custom_writer.py:151–168

(self, name)

    

    def __getattr__(self, name):
        if name in self.tb_writer_funcs:
            func = getattr(self, name, None)
            # Return a wrapper for all functions.
            def wrapper(tag, data, *args, **kwargs):
                if func is not None:
                    if name not in self.tag_mode_exceptions:
                        tag = f"{tag}/{self.mode}"
                    func(tag, data, *args, global_step=self.step, **kwargs)
    
            return wrapper
        else:
            # default __getattr__ function to get other attributes.
            try:
                attr = object.__getattr__(name)
            except AttributeError:
                raise AttributeError("type object '{}' has no attribute '{}'".format(self.selected_module, name))
            return attr