I guess I mean you could just log every node._func_name and run the log_model method if the object is of the right type