Silence pyright Optional-access on async_op=True wait() calls

PyTorch's distributed stubs type dist.barrier/all_reduce/broadcast as
returning Work | None because the sync path (async_op=False) returns
None. Pyright cannot narrow the return when async_op=True is a literal
kwarg, so every .wait(...) on the async handle flags
reportOptionalMemberAccess. Add the same # type: ignore the rest of
the codebase uses at three sites: _barrier_with_timeout, the eval-loss
broadcast, and check_nccl_health.

Author: mmshad

kempnerforge/distributed/setup.py

@@ -107,7 +107,7 @@ def _barrier_with_timeout(seconds: float, reason: str) -> None:
     """
     work = dist.barrier(async_op=True)
     try:
-        done = work.wait(timeout=timedelta(seconds=seconds))
+        done = work.wait(timeout=timedelta(seconds=seconds))  # type: ignore[reportOptionalMemberAccess]
     except RuntimeError as e:
         raise RuntimeError(
             f"Barrier timed out after {seconds}s during {reason}. "

kempnerforge/resilience/health.py

@@ -235,7 +235,7 @@ def check_nccl_health(timeout_sec: float = 10.0) -> bool:
         # timeout; older/alternate backends may return False instead. Handle
         # both so the timeout is honored regardless of version.
         try:
-            done = work.wait(timeout=timedelta(seconds=timeout_sec))
+            done = work.wait(timeout=timedelta(seconds=timeout_sec))  # type: ignore[reportOptionalMemberAccess]
         except RuntimeError as e:
             logger.warning(f"NCCL health check timed out after {timeout_sec}s: {e}")
             return False

kempnerforge/training/eval.py

@@ -93,7 +93,7 @@ def run_eval(
             async_op=True,
         )
         try:
-            done = work.wait(timeout=timedelta(seconds=_EVAL_BROADCAST_TIMEOUT_SEC))
+            done = work.wait(timeout=timedelta(seconds=_EVAL_BROADCAST_TIMEOUT_SEC))  # type: ignore[reportOptionalMemberAccess]
         except RuntimeError as e:
             logger.error(
                 f"Eval loss broadcast timed out after {_EVAL_BROADCAST_TIMEOUT_SEC}s; "