WandbLogger fixes: removed finish() from Model fit(), ValidationCallbacks fixes (#2586)

* removed finish from fit(), separate logging keys for different eval datasets

* yapf format fix

* fixed type annotations, wandb_logger existance checking in callback