Update tests for latest PL release

[rohitgr7]

Fix eco-system failures with PL latest release.

[codecov[bot]]

Codecov Report

Merging #284 (79d49f8) into master (11f2056) will not change coverage. The diff coverage is n/a.

@@          Coverage Diff          @@
##           master   #284   +/-   ##
=====================================
  Coverage      74%    74%           
=====================================
  Files          63     63           
  Lines        1474   1474           
=====================================
  Hits         1088   1088           
  Misses        386    386           

[carmocca]

CI is red

[turian]

In fact, tensorboard pins a really old version of protobuf:

# See: https://cs.opensource.google/tensorflow/tensorflow/+/master:tensorflow/tools/pip_package/setup.py?q=protobuf
protobuf >= 3.9.2, < 3.20

Related:

• https://github.com/Lightning-AI/lightning/pull/13189
• https://github.com/Lightning-AI/lightning/pull/13177
• https://github.com/Lightning-AI/lightning/issues/13159