Fix RuntimeError not caught when torchcodec fails to load (#3987)

When datasets library has torchcodec installed but FFmpeg libraries
are missing, torchcodec raises a RuntimeError during import. The
exception handler only caught ImportError and AttributeError, causing
the error to propagate and crash Unsloth imports in environments
like Colab where FFmpeg may not be installed.

Co-authored-by: Daniel Han <danielhanchen@users.noreply.github.com>
This commit is contained in:
Daniel Han 2026-02-05 06:35:10 -08:00 committed by GitHub
commit 5f9dfadb19
No known key found for this signature in database
GPG key ID: B5690EEEBB952194

View file

@ -1051,5 +1051,5 @@ def patch_torchcodec_audio_decoder():
from unsloth_zoo.dataset_utils import patch_torchcodec_audio_decoder as _patch
_patch()
except (ImportError, AttributeError):
except (ImportError, AttributeError, RuntimeError):
pass