Assertion fails when torch.mean is called with dtype argument. #2124
Description
🐞Describing the bug
- Make sure to only create an issue here for bugs in the coremltools Python package. If this is a bug with the Core ML Framework or Xcode, please submit your bug here: https://developer.apple.com/bug-reporting/
- Provide a clear and consise description of the bug.
Assertion fails when torch.mean is called with dtype argument.
Stack Trace
- If applicable, please paste the complete stack trace.
To Reproduce
- Please add a minimal code example that can reproduce the error when running it.
# Paste Python code snippet here, complete with any required import statements.
- If the model conversion succeeds, but there is a numerical mismatch in predictions, please include the code used for comparisons.
System environment (please complete the following information):
- coremltools version: 7.1
- OS (e.g. MacOS version or Linux type): macOS Sonoma 14.3
- Any other relevant version information (e.g. PyTorch or TensorFlow version):
Additional context
- Add anything else about the problem here that you want to share.