When attempting to use the `FlowEncoder` class I ran into the following error: > ```shell AttributeError: module 'nflows' has no attribute 'nn' ``` I think this is due to a missing import in `transforms.py`. Appending ```python import nflows.nn.nets ``` seems to fix the issue for now.