Skip to content

Commit b754016

Browse files
committed
Fix torch __all__
1 parent f71f826 commit b754016

File tree

1 file changed

+13
-13
lines changed

1 file changed

+13
-13
lines changed

array_api_compat/torch/_aliases.py

+13-13
Original file line numberDiff line numberDiff line change
@@ -707,18 +707,18 @@ def take(x: array, indices: array, /, *, axis: Optional[int] = None, **kwargs) -
707707
return torch.index_select(x, axis, indices, **kwargs)
708708

709709
__all__ = ['result_type', 'can_cast', 'permute_dims', 'bitwise_invert',
710-
'newaxis', 'add', 'atan2', 'bitwise_and', 'bitwise_left_shift',
711-
'bitwise_or', 'bitwise_right_shift', 'bitwise_xor', 'copysign',
712-
'divide', 'equal', 'floor_divide', 'greater', 'greater_equal',
713-
'less', 'less_equal', 'logaddexp', 'multiply', 'not_equal', 'pow',
714-
'remainder', 'subtract', 'max', 'min', 'clip', 'sort', 'prod',
715-
'sum', 'any', 'all', 'mean', 'std', 'var', 'concat', 'squeeze',
716-
'broadcast_to', 'flip', 'roll', 'nonzero', 'where', 'reshape',
717-
'arange', 'eye', 'linspace', 'full', 'ones', 'zeros', 'empty',
718-
'tril', 'triu', 'expand_dims', 'astype', 'broadcast_arrays',
719-
'UniqueAllResult', 'UniqueCountsResult', 'UniqueInverseResult',
720-
'unique_all', 'unique_counts', 'unique_inverse', 'unique_values',
721-
'matmul', 'matrix_transpose', 'vecdot', 'tensordot', 'isdtype',
722-
'take']
710+
'newaxis', 'conj', 'add', 'atan2', 'bitwise_and',
711+
'bitwise_left_shift', 'bitwise_or', 'bitwise_right_shift',
712+
'bitwise_xor', 'copysign', 'divide', 'equal', 'floor_divide',
713+
'greater', 'greater_equal', 'less', 'less_equal', 'logaddexp',
714+
'multiply', 'not_equal', 'pow', 'remainder', 'subtract', 'max',
715+
'min', 'clip', 'sort', 'prod', 'sum', 'any', 'all', 'mean', 'std',
716+
'var', 'concat', 'squeeze', 'broadcast_to', 'flip', 'roll',
717+
'nonzero', 'where', 'reshape', 'arange', 'eye', 'linspace', 'full',
718+
'ones', 'zeros', 'empty', 'tril', 'triu', 'expand_dims', 'astype',
719+
'broadcast_arrays', 'UniqueAllResult', 'UniqueCountsResult',
720+
'UniqueInverseResult', 'unique_all', 'unique_counts',
721+
'unique_inverse', 'unique_values', 'matmul', 'matrix_transpose',
722+
'vecdot', 'tensordot', 'isdtype', 'take']
723723

724724
_all_ignore = ['torch', 'get_xp']

0 commit comments

Comments
 (0)