Re-enabled forcing and moved forcing and diagnostic to masking

[clessig]

Description

Re-enabled forcing. Also moved forcing and diagnostic to masking instead of tokenizer. Some clean-up of code.

Issue Number

Closes #1762

Checklist before asking for review

• I have performed a self-review of my code
• My changes comply with basic sanity checks:
  • I have fixed formatting issues with ./scripts/actions.sh lint
  • I have run unit tests with ./scripts/actions.sh unit-test
  • I have documented my code and I have updated the docstrings.
  • I have added unit tests, if relevant
• I have tried my changes with data and code:
  • I have run the integration tests with ./scripts/actions.sh integration-test
  • (bigger changes) I have run a full training and I have written in the comment the run_id(s): launch-slurm.py --time 60
  • (bigger changes and experiments) I have shared a hegdedoc in the github issue with all the configurations and runs for this experiments
• I have informed and aligned with people impacted by my change:
  • for config changes: the MatterMost channels and/or a design doc
  • for changes of dependencies: the MatterMost software development channel

[clessig]

Test command for running inference:

uv run inference --from-run-id mi6euhax --samples 2 --options test_config.start_date=2022-10-01T00:00 test_config.end_date=2022-12-31T00:00 test_config.output.streams=[SurfaceCombined]

[clessig]

Code has been tested interactive and with 1 node / 4 DDP workers.