Fix torch.jit.load for MessagePassing modules

[rusty1s]

Fixes #8867

[codecov]

Codecov Report

All modified and coverable lines are covered by tests ✅

Comparison is base (1f93077) 89.28% compared to head (e6b258d) 89.28%.

❗ Current head e6b258d differs from pull request most recent head d85f210. Consider uploading reports for the commit d85f210 to get more accurate results

Additional details and impacted files

@@            Coverage Diff             @@
##           master    #8870      +/-   ##
==========================================
- Coverage   89.28%   89.28%   -0.01%     
==========================================
  Files         467      467              
  Lines       29907    29901       -6     
==========================================
- Hits        26703    26697       -6     
  Misses       3204     3204              

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