@@ -114,9 +114,9 @@ auto get_onedal_result_options(const py::dict& params) {
114
114
struct params2desc {
115
115
template <typename Float, typename Method, typename Task>
116
116
auto operator ()(const py::dict& params) {
117
- auto desc = dal::basic_statistics::descriptor<Float,
118
- Method, dal::basic_statistics::task::compute>()
119
- .set_result_options (get_onedal_result_options (params));
117
+ auto desc =
118
+ dal::basic_statistics::descriptor<Float, Method, dal::basic_statistics::task::compute>()
119
+ .set_result_options (get_onedal_result_options (params));
120
120
return desc;
121
121
}
122
122
};
@@ -126,54 +126,53 @@ struct params2desc_incremental {
126
126
template <typename Float, typename Method, typename Task>
127
127
auto operator ()(const py::dict& params) {
128
128
auto desc = dal::basic_statistics::descriptor<Float,
129
- dal::basic_statistics::method::dense, dal::basic_statistics::task::compute>()
130
- .set_result_options (get_onedal_result_options (params));
129
+ dal::basic_statistics::method::dense,
130
+ dal::basic_statistics::task::compute>()
131
+ .set_result_options (get_onedal_result_options (params));
131
132
return desc;
132
133
}
133
134
};
134
135
135
136
template <typename Policy, typename Task>
136
137
void init_compute_ops (py::module& m) {
137
- m.def (" compute" , [](
138
- const Policy& policy,
139
- const py::dict& params,
140
- const table& data,
141
- const table& weights) {
138
+ m.def (
139
+ " compute" ,
140
+ [](const Policy& policy, const py::dict& params, const table& data, const table& weights) {
142
141
using namespace dal ::basic_statistics;
143
142
using input_t = compute_input<Task>;
144
143
145
144
compute_ops ops (policy, input_t { data, weights }, params2desc{});
146
145
return fptype2t{ method2t{ Task{}, ops } }(params);
147
- }
148
- );
146
+ });
149
147
}
150
148
151
-
152
149
template <typename Policy, typename Task>
153
150
void init_partial_compute_ops (py::module& m) {
154
151
using prev_result_t = dal::basic_statistics::partial_compute_result<Task>;
155
- m.def (" partial_compute" , [](
156
- const Policy& policy,
157
- const py::dict& params,
158
- const prev_result_t & prev,
159
- const table& data,
160
- const table& weights) {
161
- using namespace dal ::basic_statistics;
162
- using input_t = partial_compute_input<Task>;
163
- partial_compute_ops ops (policy, input_t { prev, data, weights }, params2desc_incremental{});
164
- return fptype2t{ method2t{ Task{}, ops } }(params);
165
- }
166
- );
152
+ m.def (" partial_compute" ,
153
+ [](const Policy& policy,
154
+ const py::dict& params,
155
+ const prev_result_t & prev,
156
+ const table& data,
157
+ const table& weights) {
158
+ using namespace dal ::basic_statistics;
159
+ using input_t = partial_compute_input<Task>;
160
+ partial_compute_ops ops (policy,
161
+ input_t { prev, data, weights },
162
+ params2desc_incremental{});
163
+ return fptype2t{ method2t{ Task{}, ops } }(params);
164
+ });
167
165
}
168
166
169
167
template <typename Policy, typename Task>
170
168
void init_finalize_compute_ops (pybind11::module_& m) {
171
169
using namespace dal ::basic_statistics;
172
170
using input_t = partial_compute_result<Task>;
173
- m.def (" finalize_compute" , [](const Policy& policy, const pybind11::dict& params, const input_t & data) {
174
- finalize_compute_ops ops (policy, data, params2desc_incremental{});
175
- return fptype2t{ method2t{ Task{}, ops } }(params);
176
- });
171
+ m.def (" finalize_compute" ,
172
+ [](const Policy& policy, const pybind11::dict& params, const input_t & data) {
173
+ finalize_compute_ops ops (policy, data, params2desc_incremental{});
174
+ return fptype2t{ method2t{ Task{}, ops } }(params);
175
+ });
177
176
}
178
177
179
178
template <typename Task>
@@ -216,23 +215,28 @@ void init_partial_compute_result(py::module_& m) {
216
215
py::cast<py::object>(convert_to_pyobject (res.get_partial_max ())),
217
216
py::cast<py::object>(convert_to_pyobject (res.get_partial_sum ())),
218
217
py::cast<py::object>(convert_to_pyobject (res.get_partial_sum_squares ())),
219
- py::cast<py::object>(convert_to_pyobject (res. get_partial_sum_squares_centered ()))
220
- );
218
+ py::cast<py::object>(
219
+ convert_to_pyobject (res. get_partial_sum_squares_centered ())) );
221
220
},
222
221
[](py::tuple t) {
223
222
if (t.size () != 6 )
224
223
throw std::runtime_error (" Invalid state!" );
225
224
result_t res;
226
- if (py::cast<int >(t[0 ].attr (" size" )) != 0 ) res.set_partial_n_rows (convert_to_table (t[0 ]));
227
- if (py::cast<int >(t[1 ].attr (" size" )) != 0 ) res.set_partial_min (convert_to_table (t[1 ]));
228
- if (py::cast<int >(t[2 ].attr (" size" )) != 0 ) res.set_partial_max (convert_to_table (t[2 ]));
229
- if (py::cast<int >(t[3 ].attr (" size" )) != 0 ) res.set_partial_sum (convert_to_table (t[3 ]));
230
- if (py::cast<int >(t[4 ].attr (" size" )) != 0 ) res.set_partial_sum_squares (convert_to_table (t[4 ]));
231
- if (py::cast<int >(t[5 ].attr (" size" )) != 0 ) res.set_partial_sum_squares_centered (convert_to_table (t[5 ]));
232
-
225
+ if (py::cast<int >(t[0 ].attr (" size" )) != 0 )
226
+ res.set_partial_n_rows (convert_to_table (t[0 ]));
227
+ if (py::cast<int >(t[1 ].attr (" size" )) != 0 )
228
+ res.set_partial_min (convert_to_table (t[1 ]));
229
+ if (py::cast<int >(t[2 ].attr (" size" )) != 0 )
230
+ res.set_partial_max (convert_to_table (t[2 ]));
231
+ if (py::cast<int >(t[3 ].attr (" size" )) != 0 )
232
+ res.set_partial_sum (convert_to_table (t[3 ]));
233
+ if (py::cast<int >(t[4 ].attr (" size" )) != 0 )
234
+ res.set_partial_sum_squares (convert_to_table (t[4 ]));
235
+ if (py::cast<int >(t[5 ].attr (" size" )) != 0 )
236
+ res.set_partial_sum_squares_centered (convert_to_table (t[5 ]));
237
+
233
238
return res;
234
- }
235
- ));
239
+ }));
236
240
}
237
241
238
242
ONEDAL_PY_DECLARE_INSTANTIATOR (init_compute_result);
0 commit comments