Skip to content

Commit 8655f29

Browse files
committed
Add unique to stub file.
1 parent 4fdea81 commit 8655f29

2 files changed

Lines changed: 2 additions & 1 deletion

File tree

test/test_value_array.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -292,4 +292,4 @@ def test_unique() -> None:
292292
xs = np.array([3.7, 0.9, 0.9, 4.3, 3.7])
293293
unit = tu.MHz
294294
v_arr = xs * unit
295-
assert np.array_equal(v_arr.unique(), np.unique(xs) * unit)
295+
assert np.array_equal(v_arr.unique(), np.unique(xs) * unit)

tunits/core/__init__.pyi

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -334,6 +334,7 @@ class ValueArray(Generic[ValueType2], WithUnit):
334334
def to_proto(self, msg: tunits_pb2.ValueArray | None = None) -> tunits_pb2.ValueArray: ...
335335
def __init__(self, data: Any, unit: Any = None) -> None: ...
336336
def allclose(self, other: ValueArray | Value, *args: Any, **kwargs: float) -> bool: ...
337+
def unique(self) -> ValueArray: ...
337338
def __array__(self, dtype: DTypeLike = None) -> NDArray[Any]: ...
338339
def __array_wrap__(self, out_arr: NDArray[Any], context: Any = None) -> NDArray[Any]: ...
339340
def __copy__(self: ArrayType) -> ArrayType: ...

0 commit comments

Comments
 (0)