This is very minor, but PEP 8 prefers that you format the if test like:

```
if state.num_aux > 0:
    s1.aux = state.aux