@@ -222,32 +222,33 @@ def unops() -> list[OpsType]:
222
222
# ruff does not reformat everything. When reformatting, the
223
223
# type-ignores end up in the wrong line :/
224
224
225
- ops_info = {}
226
- # TODO add inplace ops for DataTree?
227
- ops_info ["DataTreeOpsMixin" ] = binops (other_type = "DtCompatible" ) + unops ()
228
- ops_info ["DatasetOpsMixin" ] = (
229
- binops_overload (other_type = "DsCompatible" , overload_types = ["DataTree" ])
230
- + inplace (other_type = "DsCompatible" , type_ignore = "misc" )
231
- + unops ()
232
- )
233
- ops_info ["DataArrayOpsMixin" ] = (
234
- binops_overload (other_type = "DaCompatible" , overload_types = ["Dataset" , "DataTree" ])
235
- + inplace (other_type = "DaCompatible" , type_ignore = "misc" )
236
- + unops ()
237
- )
238
- ops_info ["VariableOpsMixin" ] = (
239
- binops_overload (
240
- other_type = "VarCompatible" , overload_types = ["T_DA" , "Dataset" , "DataTree" ]
241
- )
242
- + inplace (other_type = "VarCompatible" , type_ignore = "misc" )
243
- + unops ()
244
- )
245
- ops_info ["DatasetGroupByOpsMixin" ] = binops (
246
- other_type = "Dataset | DataArray" , return_type = "Dataset"
247
- )
248
- ops_info ["DataArrayGroupByOpsMixin" ] = binops (
249
- other_type = "T_Xarray" , return_type = "T_Xarray"
250
- )
225
+ ops_info = {
226
+ # TODO add inplace ops for DataTree?
227
+ "DataTreeOpsMixin" : binops (other_type = "DtCompatible" ) + unops (),
228
+ "DatasetOpsMixin" : (
229
+ binops_overload (other_type = "DsCompatible" , overload_types = ["DataTree" ])
230
+ + inplace (other_type = "DsCompatible" , type_ignore = "misc" )
231
+ + unops ()
232
+ ),
233
+ "DataArrayOpsMixin" : (
234
+ binops_overload (
235
+ other_type = "DaCompatible" , overload_types = ["Dataset" , "DataTree" ]
236
+ )
237
+ + inplace (other_type = "DaCompatible" , type_ignore = "misc" )
238
+ + unops ()
239
+ ),
240
+ "VariableOpsMixin" : (
241
+ binops_overload (
242
+ other_type = "VarCompatible" , overload_types = ["T_DA" , "Dataset" , "DataTree" ]
243
+ )
244
+ + inplace (other_type = "VarCompatible" , type_ignore = "misc" )
245
+ + unops ()
246
+ ),
247
+ "DatasetGroupByOpsMixin" : binops (
248
+ other_type = "Dataset | DataArray" , return_type = "Dataset"
249
+ ),
250
+ "DataArrayGroupByOpsMixin" : binops (other_type = "T_Xarray" , return_type = "T_Xarray" ),
251
+ }
251
252
252
253
MODULE_PREAMBLE = '''\
253
254
"""Mixin classes with arithmetic operators."""
0 commit comments