Skip to content

Commit 63875fa

Browse files
xaduprexadupre
authored
Fix default values (#63)
1 parent c7375ca commit 63875fa

File tree

1 file changed

+81
-82
lines changed

1 file changed

+81
-82
lines changed

‎onnx_array_api/light_api/_op_vars.py‎

Lines changed: 81 additions & 82 deletions
Original file line numberDiff line numberDiff line change
@@ -10,8 +10,10 @@ def BitShift(self, direction: str = "") -> "Var":
1010
return self.make_node("BitShift", *self.vars_, direction=direction)
1111

1212
def CenterCropPad(self, axes: Optional[List[int]] = None) -> "Var":
13-
axes = axes or []
14-
return self.make_node("CenterCropPad", *self.vars_, axes=axes)
13+
kwargs = {}
14+
if axes is not None:
15+
kwargs["axes"] = axes
16+
return self.make_node("CenterCropPad", *self.vars_, **kwargs)
1517

1618
def Clip(
1719
self,
@@ -27,12 +29,14 @@ def Col2Im(
2729
pads: Optional[List[int]] = None,
2830
strides: Optional[List[int]] = None,
2931
) -> "Var":
30-
dilations = dilations or []
31-
pads = pads or []
32-
strides = strides or []
33-
return self.make_node(
34-
"Col2Im", *self.vars_, dilations=dilations, pads=pads, strides=strides
35-
)
32+
kwargs = {}
33+
if dilations is not None:
34+
kwargs["dilations"] = dilations
35+
if pads is not None:
36+
kwargs["pads"] = pads
37+
if strides is not None:
38+
kwargs["strides"] = strides
39+
return self.make_node("Col2Im", *self.vars_, **kwargs)
3640

3741
def Compress(self, axis: int = 0) -> "Var":
3842
return self.make_node("Compress", *self.vars_, axis=axis)
@@ -71,19 +75,17 @@ def ConvInteger(
7175
pads: Optional[List[int]] = None,
7276
strides: Optional[List[int]] = None,
7377
) -> "Var":
74-
dilations = dilations or []
75-
kernel_shape = kernel_shape or []
76-
pads = pads or []
77-
strides = strides or []
78+
kwargs = {}
79+
if dilations is not None:
80+
kwargs["dilations"] = dilations
81+
if kernel_shape is not None:
82+
kwargs["kernel_shape"] = kernel_shape
83+
if pads is not None:
84+
kwargs["pads"] = pads
85+
if strides is not None:
86+
kwargs["strides"] = strides
7887
return self.make_node(
79-
"ConvInteger",
80-
*self.vars_,
81-
auto_pad=auto_pad,
82-
dilations=dilations,
83-
group=group,
84-
kernel_shape=kernel_shape,
85-
pads=pads,
86-
strides=strides,
88+
"ConvInteger", *self.vars_, auto_pad=auto_pad, group=group, **kwargs
8789
)
8890

8991
def ConvTranspose(
@@ -97,23 +99,21 @@ def ConvTranspose(
9799
pads: Optional[List[int]] = None,
98100
strides: Optional[List[int]] = None,
99101
) -> "Var":
100-
dilations = dilations or []
101-
kernel_shape = kernel_shape or []
102-
output_padding = output_padding or []
103-
output_shape = output_shape or []
104-
pads = pads or []
105-
strides = strides or []
106-
return self.make_node(
107-
"ConvTranspose",
108-
*self.vars_,
109-
auto_pad=auto_pad,
110-
dilations=dilations,
111-
group=group,
112-
kernel_shape=kernel_shape,
113-
output_padding=output_padding,
114-
output_shape=output_shape,
115-
pads=pads,
116-
strides=strides,
102+
kwargs = {}
103+
if dilations is not None:
104+
kwargs["dilations"] = dilations
105+
if kernel_shape is not None:
106+
kwargs["kernel_shape"] = kernel_shape
107+
if pads is not None:
108+
kwargs["pads"] = pads
109+
if strides is not None:
110+
kwargs["strides"] = strides
111+
if output_padding is not None:
112+
kwargs["output_padding"] = output_padding
113+
if output_shape is not None:
114+
kwargs["output_shape"] = output_shape
115+
return self.make_node(
116+
"ConvTranspose", *self.vars_, auto_pad=auto_pad, group=group, **kwargs
117117
)
118118

119119
def CumSum(self, exclusive: int = 0, reverse: int = 0) -> "Var":
@@ -135,19 +135,17 @@ def DeformConv(
135135
pads: Optional[List[int]] = None,
136136
strides: Optional[List[int]] = None,
137137
) -> "Var":
138-
dilations = dilations or []
139-
kernel_shape = kernel_shape or []
140-
pads = pads or []
141-
strides = strides or []
138+
kwargs = {}
139+
if dilations is not None:
140+
kwargs["dilations"] = dilations
141+
if kernel_shape is not None:
142+
kwargs["kernel_shape"] = kernel_shape
143+
if pads is not None:
144+
kwargs["pads"] = pads
145+
if strides is not None:
146+
kwargs["strides"] = strides
142147
return self.make_node(
143-
"DeformConv",
144-
*self.vars_,
145-
dilations=dilations,
146-
group=group,
147-
kernel_shape=kernel_shape,
148-
offset_group=offset_group,
149-
pads=pads,
150-
strides=strides,
148+
"DeformConv", *self.vars_, group=group, offset_group=offset_group, **kwargs
151149
)
152150

153151
def DequantizeLinear(self, axis: int = 1) -> "Var":
@@ -204,12 +202,11 @@ def MatMulInteger(
204202
def MaxRoiPool(
205203
self, pooled_shape: Optional[List[int]] = None, spatial_scale: float = 1.0
206204
) -> "Var":
207-
pooled_shape = pooled_shape or []
205+
kwargs = {}
206+
if pooled_shape is not None:
207+
kwargs["pooled_shape"] = pooled_shape
208208
return self.make_node(
209-
"MaxRoiPool",
210-
*self.vars_,
211-
pooled_shape=pooled_shape,
212-
spatial_scale=spatial_scale,
209+
"MaxRoiPool", *self.vars_, spatial_scale=spatial_scale, **kwargs
213210
)
214211

215212
def MaxUnpool(
@@ -218,16 +215,14 @@ def MaxUnpool(
218215
pads: Optional[List[int]] = None,
219216
strides: Optional[List[int]] = None,
220217
) -> "Var":
221-
kernel_shape = kernel_shape or []
222-
pads = pads or []
223-
strides = strides or []
224-
return self.make_node(
225-
"MaxUnpool",
226-
*self.vars_,
227-
kernel_shape=kernel_shape,
228-
pads=pads,
229-
strides=strides,
230-
)
218+
kwargs = {}
219+
if kernel_shape is not None:
220+
kwargs["kernel_shape"] = kernel_shape
221+
if pads is not None:
222+
kwargs["pads"] = pads
223+
if strides is not None:
224+
kwargs["strides"] = strides
225+
return self.make_node("MaxUnpool", *self.vars_, **kwargs)
231226

232227
def MelWeightMatrix(self, output_datatype: int = 1) -> "Var":
233228
return self.make_node(
@@ -267,19 +262,17 @@ def QLinearConv(
267262
pads: Optional[List[int]] = None,
268263
strides: Optional[List[int]] = None,
269264
) -> "Var":
270-
dilations = dilations or []
271-
kernel_shape = kernel_shape or []
272-
pads = pads or []
273-
strides = strides or []
265+
kwargs = {}
266+
if kernel_shape is not None:
267+
kwargs["kernel_shape"] = kernel_shape
268+
if pads is not None:
269+
kwargs["pads"] = pads
270+
if strides is not None:
271+
kwargs["strides"] = strides
272+
if dilations is not None:
273+
kwargs["dilations"] = dilations
274274
return self.make_node(
275-
"QLinearConv",
276-
*self.vars_,
277-
auto_pad=auto_pad,
278-
dilations=dilations,
279-
group=group,
280-
kernel_shape=kernel_shape,
281-
pads=pads,
282-
strides=strides,
275+
"QLinearConv", *self.vars_, auto_pad=auto_pad, group=group, **kwargs
283276
)
284277

285278
def QLinearMatMul(
@@ -303,15 +296,17 @@ def RandomNormal(
303296
seed: float = 0.0,
304297
shape: Optional[List[int]] = None,
305298
) -> "Var":
306-
shape = shape or []
299+
kwargs = {}
300+
if shape is not None:
301+
kwargs["shape"] = shape
307302
return self.make_node(
308303
"RandomNormal",
309304
*self.vars_,
310305
dtype=dtype,
311306
mean=mean,
312307
scale=scale,
313308
seed=seed,
314-
shape=shape,
309+
**kwargs,
315310
)
316311

317312
def RandomUniform(
@@ -322,15 +317,17 @@ def RandomUniform(
322317
seed: float = 0.0,
323318
shape: Optional[List[int]] = None,
324319
) -> "Var":
325-
shape = shape or []
320+
kwargs = {}
321+
if shape is not None:
322+
kwargs["shape"] = shape
326323
return self.make_node(
327324
"RandomUniform",
328325
*self.vars_,
329326
dtype=dtype,
330327
high=high,
331328
low=low,
332329
seed=seed,
333-
shape=shape,
330+
**kwargs,
334331
)
335332

336333
def Range(
@@ -437,19 +434,21 @@ def Resize(
437434
mode: str = "nearest",
438435
nearest_mode: str = "round_prefer_floor",
439436
) -> "Var":
440-
axes = axes or []
437+
kwargs = {}
438+
if axes is not None:
439+
kwargs["axes"] = axes
441440
return self.make_node(
442441
"Resize",
443442
*self.vars_,
444443
antialias=antialias,
445-
axes=axes,
446444
coordinate_transformation_mode=coordinate_transformation_mode,
447445
cubic_coeff_a=cubic_coeff_a,
448446
exclude_outside=exclude_outside,
449447
extrapolation_value=extrapolation_value,
450448
keep_aspect_ratio_policy=keep_aspect_ratio_policy,
451449
mode=mode,
452450
nearest_mode=nearest_mode,
451+
**kwargs,
453452
)
454453

455454
def RoiAlign(

0 commit comments

Comments
 (0)