Skip to content

Commit 1a33d7c

Browse files
committed
Fix __array_namespace_info__ stub not registering
1 parent bc1e37e commit 1a33d7c

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

array_api_tests/stubs.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -44,7 +44,7 @@
4444

4545
category_to_funcs: Dict[str, List[FunctionType]] = {}
4646
for name, mod in name_to_mod.items():
47-
if name.endswith("_functions"):
47+
if name.endswith("_functions") or name == "info": # info functions file just named info.py
4848
category = name.replace("_functions", "")
4949
objects = [getattr(mod, name) for name in mod.__all__]
5050
assert all(isinstance(o, FunctionType) for o in objects) # sanity check

0 commit comments

Comments
 (0)