File tree Expand file tree Collapse file tree 1 file changed +7
-3
lines changed Expand file tree Collapse file tree 1 file changed +7
-3
lines changed Original file line number Diff line number Diff line change 25
25
26
26
#if ULAB_SUPPORTS_COMPLEX
27
27
28
+ //| import builtins
29
+ //|
28
30
//| import ulab.numpy
29
31
30
- //| def real(val) :
32
+ //| def real(val: ulab.numpy.ndarray) -> ulab.numpy.ndarray :
31
33
//| """
32
34
//| Return the real part of the complex argument, which can be
33
35
//| either an ndarray, or a scalar."""
@@ -54,7 +56,7 @@ mp_obj_t carray_real(mp_obj_t _source) {
54
56
55
57
MP_DEFINE_CONST_FUN_OBJ_1 (carray_real_obj , carray_real );
56
58
57
- //| def imag(val) :
59
+ //| def imag(val: ulab.numpy.ndarray) -> ulab.numpy.ndarray :
58
60
//| """
59
61
//| Return the imaginary part of the complex argument, which can be
60
62
//| either an ndarray, or a scalar."""
@@ -82,7 +84,9 @@ MP_DEFINE_CONST_FUN_OBJ_1(carray_imag_obj, carray_imag);
82
84
83
85
#if ULAB_NUMPY_HAS_CONJUGATE
84
86
85
- //| def conjugate(val):
87
+ //| def conjugate(
88
+ //| val: builtins.complex | ulab.numpy.ndarray
89
+ //| ) -> builtins.complex | ulab.numpy.ndarray:
86
90
//| """
87
91
//| Return the conjugate of the complex argument, which can be
88
92
//| either an ndarray, or a scalar."""
You can’t perform that action at this time.
0 commit comments