Need to create the torch Generator on the correct device.

[drewoldag]

The torch generator will, by default be created on CPU, but it needs to be in the same place as the rest of the data. On Mac, that means MPS by default.

[drewoldag]

Given that this addresses the issue, we should draft a new release after this merges.

[Copilot]

Pull request overview

This PR updates Hyrax’s Ignite distributed dataloader helper to construct the torch.Generator on the Ignite-selected device (intended to align RNG behavior with non-CPU backends like MPS on macOS).

Changes:

• Create torch.Generator with device=idist.device() inside dist_data_loader.

[codecov]

Codecov Report

✅ All modified and coverable lines are covered by tests.
✅ Project coverage is 67.72%. Comparing base (7d52a5f) to head (491cd52).

Additional details and impacted files

@@           Coverage Diff           @@
##             main     #924   +/-   ##
=======================================
  Coverage   67.72%   67.72%           
=======================================
  Files          70       70           
  Lines        6805     6805           
=======================================
  Hits         4609     4609           
  Misses       2196     2196           

☔ View full report in Codecov by Sentry.
📢 Have feedback on the report? Share it here.

🚀 New features to boost your workflow:

• ❄️ Test Analytics: Detect flaky tests, report on failures, and find test suite problems.
• 📦 JS Bundle Analysis: Save yourself from yourself by tracking and limiting bundle sizes in JS merges.

[mtauraso]

Works on my machine