Decorated functions lead to error

```
import contextlib

@contextlib.contextmanager
def some_helper():
    pass
```

produces

```
snakefmt.exceptions.InvalidPython: Got error:
```
Cannot parse: 41:1: @
```
while formatting code with black.
```