fix: logger accepts arrays as data and use tensorboard_logger Logger

[Your-Cheese]

What?

I was running experiments with MuZero when I ran into some errors with the loggers. I made changes so that the logger accepts Jax arrays as data to log and uses a tensorboard_logger Logger object instead of a global variable.

Why?

Somehow, a Jax array gets passed as input to the logger every so often, resulting in an error.
When using Hydra to perform multiple runs with the Tensorboard logger enabled, an error occurs on the second run due to the logger from the previous run existing and already being configured because it is based on a global variable in the tensorboard_logger library.

How?

In the case that a Jax array gets passed to the logger, it converts it to a scalar.
By using an instance of the Logger class instead of a global variable, the Tensorboard logger properly goes out of scope between runs.

Extra

I also had issues with running the commit linter step when making the commit because it would say "Please add rules to your commitlint.config.js". Making this change to the dependencies resolved it. I'm not sure if it's an issue on my end or not, although I did follow the steps for installing the pre-commit hooks.

[EdanToledo]

Hello, thanks so much for spotting this and offering a fix. I'm relatively happy with the tensorboard change however this issue of a jax array being passed seems as if it would be common for all loggers. In that case, we should be making a fix in the StoixLogger class as that filters down to all the loggers being used. That way, the fix will be present for all the other loggers. If you're happy to make changes to this PR, I'd like the following:

1. Remove the if statements and .item() calls from the individual logger.
2. Perform a check like this after the np.mean or the "describe" function call in the stoix logger
3. check to see if the problem is fixed.

[Your-Cheese]

I really appreciate the feedback and being so specific on what to change. I moved the check for jax arrays to the StoixLogger log function, and it seems to work fine.

[EdanToledo]

Thanks so much for the changes. Once the tests pass, I'll merge.

[EdanToledo]

LGTM!