Capture more test dependencies in constraints

[booxter]

This is going to be a follow-up to: #558

I've noticed that more dependencies are not constrained because they are not mentioned in requirements files. E.g. wandb and tensorboard (there are more). Some are capped (mypy) in tox.ini. All of them should be in a requirements file that is included when calculating constraints, to avoid unexpected breakages if new versions of these dependencies are released.

Splitting this out into a separate PR so that we don't boil the ocean and can land main constraints PR quicker.