wandb_logger.watch(log_graph=True/False) not working

wandb_logger.watch(model, log_graph=False) is not working.

Error : watch() got an unexpected keyword argument ‘log_graph’