coremltools-8.0b2 requires torchvision

[atiorh]

🐞Describing the bug

It looks like coremltools now requires torchvision. Is this expected?

Stack Trace

Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
  File "conda/lib/python3.11/site-packages/coremltools/__init__.py", line 97, in <module>
    from . import converters, models, optimize, proto
...
  File ".conda/lib/python3.11/site-packages/coremltools/optimize/torch/_utils/graph_utils.py", line 14, in <module>
    import torchvision as _torchvision
ModuleNotFoundError: No module named 'torchvision'

To Reproduce

conda create -n repro python=3.11 -y && conda activate && pip install coremltools==8.0b2 && python -c "import coremltools as ct"

System environment (please complete the following information):

• coremltools version: 8.0b2
• OS (e.g. MacOS version or Linux type): 24B5024e
• Any other relevant version information (e.g. PyTorch or TensorFlow version): Python 3.11.9

[DawerG]

Thanks for reporting. It's not expected, we are working on addressing this.