Update _capture_strategies.py

justinchuby · web-flow · commit f422da97fbb9 · 2025-09-05T09:24:20.000-07:00
diff --git a/torch/onnx/_internal/exporter/_capture_strategies.py b/torch/onnx/_internal/exporter/_capture_strategies.py
@@ -280,7 +280,7 @@ def _failure(self, model, e) -> None:
         )
 
 
-CAPTURE_STRATEGIES: tuple[CaptureStrategy, ...] = (
+CAPTURE_STRATEGIES: tuple[type[CaptureStrategy], ...] = (
     TorchExportNonStrictStrategy,  # strict=False is preferred over strict=True because it does not have dynamo issues
     TorchExportStrictStrategy,
 )
