Currently, test_wandb was never added to either of the CI builds and some of its tests are failing locally. So, we need to fix them and make sure they pass on the CI builds too.