The `core.test_gamenet.TestGAMENet.test_loss_is_finite` unit test is unstable

[EricSchrock]

While developing #788, I saw the core.test_gamenet.TestGAMENet.test_loss_is_finite unit test fail during a GitHub actions run, though I had not changed any related files. I reran the pipeline without changes, and it passed. Opening an issue ticket for awareness, in case others are intermittently seeing this issue too.

======================================================================
FAIL: test_loss_is_finite (core.test_gamenet.TestGAMENet.test_loss_is_finite)
----------------------------------------------------------------------
Traceback (most recent call last):
  File "/home/runner/work/PyHealth/PyHealth/tests/core/test_gamenet.py", line 111, in test_loss_is_finite
    self.assertTrue(torch.isfinite(ret["loss"]).all())
    ~~~~~~~~~~~~~~~^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
AssertionError: tensor(False) is not true

https://github.com/EricSchrock/PyHealth/actions/runs/21098985432/attempts/1