pytorch - 💡(How to fix) Fix Make `ONNX_DEFAULT_OPSET` public. [4 comments, 2 participants]

Fix Plan

To make the ONNX_DEFAULT_OPSET a public constant, we can create a new public module or add it to an existing one.

Steps

• Create a new file torch/onnx/constants.py with the following content:

from torch.onnx._constants import ONNX_DEFAULT_OPSET as _ONNX_DEFAULT_OPSET

ONNX_DEFAULT_OPSET = _ONNX_DEFAULT_OPSET

• Alternatively, add the constant to an existing public module, such as torch/onnx/__init__.py:

from torch.onnx._constants import ONNX_DEFAULT_OPSET

__all__ = [..., 'ONNX_DEFAULT_OPSET']

• Update the documentation to reflect the new public constant.

Verification

To verify the fix, check that the constant is accessible from the new public module:

import torch.onnx

print(torch.onnx.ONNX_DEFAULT_OPSET)

This should print the default opset version.

Extra Tips

• Make sure to update the documentation and tests to reflect the new public constant.
• Consider adding a deprecation warning for the private constant to encourage users to switch to the new public one.