fix(wandb): allow use of sweeps

[borisdayma]

Before submitting

• Was this discussed/approved via a Github issue? (no need for typos and docs improvements)
• Did you read the contributor guideline, Pull Request section?
• Did you make sure to update the docs?
  not applicable
• Did you write any new necessary tests?
• If you made a notable change (that affects users), did you update the CHANGELOG?

What does this PR do?

Fixes #1290 .

PR review

Anyone in the community is free to review the PR once the tests have passed.
If we didn't discuss your PR in Github issues there's a high chance it will not be merged.

Did you have fun?

Make sure you had fun coding 🙃

[codecov]

Codecov Report

Merging #1512 into master will not change coverage.
The diff coverage is 100%.

@@          Coverage Diff           @@
##           master   #1512   +/-   ##
======================================
  Coverage      88%     88%           
======================================
  Files          68      68           
  Lines        3956    3956           
======================================
  Hits         3501    3501           
  Misses        455     455           

[Borda]

pls add a test for it

[borisdayma]

The test has been updated but the way it's set up, we only see that the function has been called -> see here

This is due to the fact that wandb uses context and it does not work well with pytest (there was some errors with some platforms tested such as windows).

Similar behavior happens with other loggers which also use mocks and verify only that the correct function has been called.

Adding another test overwriting a previous value would only check the same function has been called (so no more than current test scope).

[Borda]

LGTM 🐰

[mergify]

This pull request is now in conflict... :(