Raise ImportError on spec failure

scotts · scotts · commit 799307051da6 · 2025-03-17T12:02:33.000-07:00
diff --git a/src/torchcodec/decoders/_core/ops.py b/src/torchcodec/decoders/_core/ops.py
@@ -44,6 +44,8 @@ def load_torchcodec_extension():
                 pybind_ops_module_name,
                 _get_extension_path(pybind_ops_library_name),
             )
+            if spec is None:
+                raise ImportError("Unable to load spec for pybind_ops")
 
             global _pybind_ops
             _pybind_ops = importlib.util.module_from_spec(spec)

Original file line number	Diff line number	Diff line change
`@@ -44,6 +44,8 @@ def load_torchcodec_extension():`
`44`	`44`	`pybind_ops_module_name,`
`45`	`45`	`_get_extension_path(pybind_ops_library_name),`
`46`	`46`	`)`
	`47`	`+ if spec is None:`
	`48`	`+ raise ImportError("Unable to load spec for pybind_ops")`
`47`	`49`
`48`	`50`	`global _pybind_ops`
`49`	`51`	`_pybind_ops = importlib.util.module_from_spec(spec)`