Couldn't make Wandb run on a TPU

I tried a lot of things but was not able to run Wandb on TPU.

Kernel

If there is something wrong in my code structure please let me know :slight_smile: . Have raised an issue on github as well.

Cheers!

Thanks Harveen!

The conversation is continued Here