fix(integrations): fix TF compatibility issues with WandbModelCheckpoint

[soumik12345]

Fixes GROWTH2-109

Description

This PR fixes compatibility issue of WandbModelCheckpoint with TF version < 2.6.0.

Co-author: @ayulockin

Testing

How was this PR tested?

Checklist

• Include reference to internal ticket "Fixes WB-NNNN" and/or GitHub issue "Fixes #NNNN" (if applicable)
• Ensure PR title compliance with the conventional commits standards

[codecov]

Codecov Report

Merging #4432 (423b075) into main (48f9258) will decrease coverage by 0.04%.
The diff coverage is 83.33%.

Additional details and impacted files

@@            Coverage Diff             @@
##             main    #4432      +/-   ##
==========================================
- Coverage   83.02%   82.98%   -0.05%     
==========================================
  Files         261      261              
  Lines       33242    33257      +15     
==========================================
- Hits        27600    27599       -1     
- Misses       5642     5658      +16     

Flag	Coverage Δ
functest	56.65% <83.33%> (+0.01%)	⬆️
unittest	73.19% <16.66%> (-0.09%)	⬇️

Flags with carried forward coverage won't be shown. Click here to find out more.

Impacted Files	Coverage Δ
...db/integration/keras/callbacks/model_checkpoint.py	72.97% <83.33%> (-10.08%)	⬇️
wandb/integration/tensorboard/monkeypatch.py	91.35% <0.00%> (-2.47%)	⬇️
wandb/filesync/step_prepare.py	93.67% <0.00%> (-1.27%)	⬇️
wandb/sdk/interface/interface_shared.py	86.56% <0.00%> (-0.78%)	⬇️
wandb/sdk/internal/internal_api.py	89.17% <0.00%> (-0.52%)	⬇️
wandb/sdk/wandb_setup.py	88.44% <0.00%> (-0.51%)	⬇️
wandb/sdk/lib/mailbox.py	93.15% <0.00%> (-0.35%)	⬇️
wandb/sdk/wandb_artifacts.py	83.61% <0.00%> (-0.11%)	⬇️
wandb/cli/cli.py	69.40% <0.00%> (-0.10%)	⬇️
... and 3 more

[dmitryduev]

One nit, otherwise LGTM. Thanks!

[ayulockin]

Hey @dmitryduev is this PR ready to be merged? :)

[dmitryduev]

Thanks @ayulockin - please see my comments.

[dmitryduev]

the name is a bit misleading as you're not patching anything... I'd do something like
self._is_old_tf_keras_version: Optional[bool] = None
in init
then
convert this one into a property called is_old_tf_keras_version that sets the private self._is_old_tf_keras_version on first invocation and then just returns it -- there's no need to recalculate it every time.

[dmitryduev]

caching was not fixed, I added it in 67756e1

[ayulockin]

Hey @dmitryduev I made the changes as per your feedback.

However, I am not sure if I did this correctly: "You need to add this new shard to both tox.ini and Circle config.yml for the tests to be run."

Also mypy is passing locally but it's failing here.

[ayulockin]

Hey @dmitryduev the code-check is failing because of this: ERROR: InvocationError for command /mnt/ramdisk/.tox/codecovcheck/bin/python tools/coverage-tool.py check (exited with code 1). What's missing to fix this?

I am getting the same InvocationError when testing locally.