Skip to content

Commit f7162f8

Browse files
committed
Address review comments by @mtsokol.
1 parent 018c567 commit f7162f8

File tree

1 file changed

+3
-2
lines changed

1 file changed

+3
-2
lines changed

sparse/mlir_backend/_array.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,6 @@
11
import numpy as np
22

3+
from ._dtypes import DType
34
from .levels import StorageFormat
45

56

@@ -21,7 +22,7 @@ def ndim(self) -> int:
2122
return len(self.shape)
2223

2324
@property
24-
def dtype(self):
25+
def dtype(self) -> type[DType]:
2526
return self._storage.get_storage_format().dtype
2627

2728
@property
@@ -34,7 +35,7 @@ def _get_mlir_type(self):
3435
def _to_module_arg(self):
3536
return self._storage.to_module_arg()
3637

37-
def copy(self):
38+
def copy(self) -> "Array":
3839
from ._conversions import from_constituent_arrays
3940

4041
arrs = tuple(arr.copy() for arr in self.get_constituent_arrays())

0 commit comments

Comments
 (0)