File tree Expand file tree Collapse file tree 1 file changed +8
-2
lines changed Expand file tree Collapse file tree 1 file changed +8
-2
lines changed Original file line number Diff line number Diff line change @@ -343,8 +343,14 @@ def apply(self, model):
343
343
used_configurations += [node .name ]
344
344
345
345
# set specified defaults
346
- default_configs = {k : v for k , v in model_config ["Defaults" ].items () if k not in model_config }
347
- default_configs = {k : v [0 ] for k , v in default_configs .items () if v [1 ] == "all" or node .op_type in v [1 ]}
346
+ default_values = []
347
+ for key , value in model_config ["Defaults" ].items ():
348
+ assert len (value ) % 2 == 0
349
+ if key not in model_config :
350
+ for val , op in zip (value [::2 ], value [1 ::2 ]):
351
+ default_values .append ((key , val , op ))
352
+ assert not (op == "all" and len (value ) > 2 )
353
+ default_configs = {key : val for key , val , op in default_values if op == "all" or node .op_type in op }
348
354
for attr , value in default_configs .items ():
349
355
inst .set_nodeattr (attr , value )
350
356
You can’t perform that action at this time.
0 commit comments