Fix warning when running integration test

[rusheb]

The integration test produces the following warning:

tests/integration/test_train_model.py::test_train
/Users/rusheb/code/maze-transformer/venv/lib/python3.10/site-packages/torch/utils/data/dataloader.py:554: UserWarning: This DataLoader will create 16 worker processes in total. Our suggested max number of worker in current system is 8 (cpuset is not taken into account), which is smaller than what this DataLoader is going to create. Please be aware that excessive worker creation might get DataLoader running slow or even freeze, lower the worker number to avoid potential slowness/freeze if necessary.
warnings.warn(_create_warning_msg(

Docs: https://docs.pytest.org/en/stable/how-to/capture-warnings.html

We should fix it by reducing num_workers in the training config

[cmathw]

Addressed in this PR: change num_workers #103 @rusheb Can I move this to done?

[rusheb]

Be my guest! Thanks for the fix!