Support: Pytest Squim Issue Locally

[ibevers]

Issue Description

Environment Details

MacOS, Python3.12

Attempted Solutions

poetry cache clear --all pypi
poetry update

Reproduction Steps

Run poetry run pytest

Additional Notes

No response

[ibevers]

Here is what happens when I install a clean senselab and run tests:
https://gist.githubusercontent.com/ibevers/6a8aacf3b2be2ef362f836cbf5c27607/raw/20b8762e87edde71a88cb98e438c74626aeed8e5/gistfile1.txt

[fabiocat93]

Thank you @ibevers for bringing up this issue! This was already discussed internally (Slack) yesterday, and maybe you found a solution already. If not, I would recommend trying the following steps to address the problem:

1. Install with Poetry without cache:
   poetry install --no-cache --with dev

2. Debug with a smaller test so that you isolate the problematic function:

from torchaudio.pipelines import SQUIM_SUBJECTIVE
model = SQUIM_SUBJECTIVE.get_model()

[ibevers]

@fabiocat93 Thanks, I will try that:)

[fabiocat93]

hi @ibevers , is this still an open issue for you? How did your isolated experiment with SQUIM_SUBJECTIVE go?

[ibevers]

@fabiocat93 doesn't work:

>>> from torchaudio.pipelines import SQUIM_SUBJECTIVE
model = SQUIM_SUBJECTIVE.get_model()>>> model = SQUIM_SUBJECTIVE.get_model()
squim_subjective_bvcc_daps.pth
Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
  File "/Users/isaacbevers/Library/Caches/pypoetry/virtualenvs/senselab-_dRIpWVy-py3.12/lib/python3.12/site-packages/torchaudio/pipelines/_squim_pipeline.py", line 130, in get_model
    state_dict = torch.load(path, weights_only=True)
                 ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
  File "/Users/isaacbevers/Library/Caches/pypoetry/virtualenvs/senselab-_dRIpWVy-py3.12/lib/python3.12/site-packages/torch/serialization.py", line 1326, in load
    with _open_zipfile_reader(opened_file) as opened_zipfile:
         ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
  File "/Users/isaacbevers/Library/Caches/pypoetry/virtualenvs/senselab-_dRIpWVy-py3.12/lib/python3.12/site-packages/torch/serialization.py", line 671, in __init__
    super().__init__(torch._C.PyTorchFileReader(name_or_buffer))
                     ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
RuntimeError: PytorchStreamReader failed reading zip archive: failed finding central directory

[ibevers]

@fabiocat93 Still open

[fabiocat93]

@fabiocat93 doesn't work:

>>> from torchaudio.pipelines import SQUIM_SUBJECTIVE
model = SQUIM_SUBJECTIVE.get_model()>>> model = SQUIM_SUBJECTIVE.get_model()
squim_subjective_bvcc_daps.pth
Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
  File "/Users/isaacbevers/Library/Caches/pypoetry/virtualenvs/senselab-_dRIpWVy-py3.12/lib/python3.12/site-packages/torchaudio/pipelines/_squim_pipeline.py", line 130, in get_model
    state_dict = torch.load(path, weights_only=True)
                 ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
  File "/Users/isaacbevers/Library/Caches/pypoetry/virtualenvs/senselab-_dRIpWVy-py3.12/lib/python3.12/site-packages/torch/serialization.py", line 1326, in load
    with _open_zipfile_reader(opened_file) as opened_zipfile:
         ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
  File "/Users/isaacbevers/Library/Caches/pypoetry/virtualenvs/senselab-_dRIpWVy-py3.12/lib/python3.12/site-packages/torch/serialization.py", line 671, in __init__
    super().__init__(torch._C.PyTorchFileReader(name_or_buffer))
                     ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
RuntimeError: PytorchStreamReader failed reading zip archive: failed finding central directory

This suggests that the issue isn’t with senselab itself, which makes sense since others aren’t encountering it. The focus now is on understanding what might be causing it in your case