We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 9ae5083 commit a99bd76Copy full SHA for a99bd76
1 file changed
array_api_strict/_info.py
@@ -79,7 +79,7 @@ def dtypes(
79
elif isinstance(kind, tuple):
80
res: DataTypes = {}
81
for k in kind:
82
- res.update(self.dtypes(kind=k))
+ res.update(self.dtypes(kind=k, device=device))
83
return res
84
raise ValueError(f"unsupported kind: {kind!r}")
85
0 commit comments