Is anyone looking at this?
add_graph is defined as -
def add_graph(self, model, input_to_model=None, verbose=False):
So this states that the second argument input_to_model is optional, so when we do not provide any value to it, it throws an error.
Is this a bug? Should it be raised somewhere else?