Add missing functions in _array_module._top_level_attrs

honno · honno · commit c76aaee2d0c2 · 2023-05-02T12:54:11.000+01:00
diff --git a/array_api_tests/_array_module.py b/array_api_tests/_array_module.py
@@ -62,7 +62,7 @@ def __repr__(self):
 ]
 _constants = ["e", "inf", "nan", "pi"]
 _funcs = [f.__name__ for funcs in stubs.category_to_funcs.values() for f in funcs]
-_funcs += ["take", "isdtype"]  # TODO: bump spec and update array-api-tests to new spec layout
+_funcs += ["take", "isdtype", "conj", "imag", "real"]  # TODO: bump spec and update array-api-tests to new spec layout
 _top_level_attrs = _dtypes + _constants + _funcs + stubs.EXTENSIONS
 
 for attr in _top_level_attrs:

Original file line number	Diff line number	Diff line change
`@@ -62,7 +62,7 @@ def __repr__(self):`
`62`	`62`	`]`
`63`	`63`	`_constants = ["e", "inf", "nan", "pi"]`
`64`	`64`	`_funcs = [f.__name__ for funcs in stubs.category_to_funcs.values() for f in funcs]`
`65`		`-_funcs += ["take", "isdtype"] # TODO: bump spec and update array-api-tests to new spec layout`
	`65`	`+_funcs += ["take", "isdtype", "conj", "imag", "real"] # TODO: bump spec and update array-api-tests to new spec layout`
`66`	`66`	`_top_level_attrs = _dtypes + _constants + _funcs + stubs.EXTENSIONS`
`67`	`67`
`68`	`68`	`for attr in _top_level_attrs:`