Skip to content

Commit a06060a

Browse files
committed
fix extension finder
1 parent 8c0bc38 commit a06060a

1 file changed

Lines changed: 3 additions & 1 deletion

File tree

src/torchcodec/_internally_replaced_utils.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -18,9 +18,11 @@ def _get_extension_path(lib_name: str) -> str:
1818
extension_suffixes = importlib.machinery.EXTENSION_SUFFIXES
1919
elif sys.platform == "darwin":
2020
extension_suffixes = importlib.machinery.EXTENSION_SUFFIXES + [".dylib"]
21+
elif sys.platform in ("win32", "cygwin"):
22+
extension_suffixes = importlib.machinery.EXTENSION_SUFFIXES + [".dll"]
2123
else:
2224
raise NotImplementedError(
23-
"Platforms other than linux/darwin are not supported yet"
25+
f"{sys.platform = } is not not supported"
2426
)
2527
loader_details = (
2628
importlib.machinery.ExtensionFileLoader,

0 commit comments

Comments
 (0)