@@ -71,21 +71,21 @@ cdef class DistanceMetric{{name_suffix}}(DistanceMetric):
71
71
cdef object func
72
72
cdef object kwargs
73
73
74
- cdef float64_t dist(
74
+ cdef {{INPUT_DTYPE_t}} dist(
75
75
self,
76
76
const {{INPUT_DTYPE_t}}* x1,
77
77
const {{INPUT_DTYPE_t}}* x2,
78
78
intp_t size,
79
79
) except -1 nogil
80
80
81
- cdef float64_t rdist(
81
+ cdef {{INPUT_DTYPE_t}} rdist(
82
82
self,
83
83
const {{INPUT_DTYPE_t}}* x1,
84
84
const {{INPUT_DTYPE_t}}* x2,
85
85
intp_t size,
86
86
) except -1 nogil
87
87
88
- cdef float64_t dist_csr(
88
+ cdef {{INPUT_DTYPE_t}} dist_csr(
89
89
self,
90
90
const {{INPUT_DTYPE_t}}* x1_data,
91
91
const int32_t* x1_indices,
@@ -98,7 +98,7 @@ cdef class DistanceMetric{{name_suffix}}(DistanceMetric):
98
98
const intp_t size,
99
99
) except -1 nogil
100
100
101
- cdef float64_t rdist_csr(
101
+ cdef {{INPUT_DTYPE_t}} rdist_csr(
102
102
self,
103
103
const {{INPUT_DTYPE_t}}* x1_data,
104
104
const int32_t* x1_indices,
@@ -114,14 +114,14 @@ cdef class DistanceMetric{{name_suffix}}(DistanceMetric):
114
114
cdef int pdist(
115
115
self,
116
116
const {{INPUT_DTYPE_t}}[:, ::1] X,
117
- float64_t [:, ::1] D,
117
+ {{INPUT_DTYPE_t}} [:, ::1] D,
118
118
) except -1
119
119
120
120
cdef int cdist(
121
121
self,
122
122
const {{INPUT_DTYPE_t}}[:, ::1] X,
123
123
const {{INPUT_DTYPE_t}}[:, ::1] Y,
124
- float64_t [:, ::1] D,
124
+ {{INPUT_DTYPE_t}} [:, ::1] D,
125
125
) except -1
126
126
127
127
cdef int pdist_csr(
@@ -130,7 +130,7 @@ cdef class DistanceMetric{{name_suffix}}(DistanceMetric):
130
130
const int32_t[::1] x1_indices,
131
131
const int32_t[::1] x1_indptr,
132
132
const intp_t size,
133
- float64_t [:, ::1] D,
133
+ {{INPUT_DTYPE_t}} [:, ::1] D,
134
134
) except -1 nogil
135
135
136
136
cdef int cdist_csr(
@@ -142,11 +142,11 @@ cdef class DistanceMetric{{name_suffix}}(DistanceMetric):
142
142
const int32_t[::1] x2_indices,
143
143
const int32_t[::1] x2_indptr,
144
144
const intp_t size,
145
- float64_t [:, ::1] D,
145
+ {{INPUT_DTYPE_t}} [:, ::1] D,
146
146
) except -1 nogil
147
147
148
- cdef float64_t _rdist_to_dist(self, {{INPUT_DTYPE_t}} rdist) except -1 nogil
148
+ cdef {{INPUT_DTYPE_t}} _rdist_to_dist(self, {{INPUT_DTYPE_t}} rdist) except -1 nogil
149
149
150
- cdef float64_t _dist_to_rdist(self, {{INPUT_DTYPE_t}} dist) except -1 nogil
150
+ cdef {{INPUT_DTYPE_t}} _dist_to_rdist(self, {{INPUT_DTYPE_t}} dist) except -1 nogil
151
151
152
152
{{endfor}}