This PR fixes circular import error arising from importing UNet in models/__init__.py.
Specifically, the error was arising because pre-commit hooks were reordering import statements in models/__init__.py in a lexicographic order, putting import of UNet after import of model_factory, which is also importing UNet.
What: Removed imports of UNet and LadderVAE modules from models/__init__.py, mirrored changes in impacted scripts.
Why: It avoids circular imports and makes importing UNet and LadderVAE more explicit.
Please ensure your PR meets the following requirements:
[ ] Code builds and passes tests locally, including doctests
[ ] New tests have been added (for bug fixes/features)
[ ] Pre-commit passes
[ ] PR to the documentation exists (for bug fixes / features)
Description
This PR fixes circular import error arising from importing
UNetinmodels/__init__.py. Specifically, the error was arising because pre-commit hooks were reordering import statements inmodels/__init__.pyin a lexicographic order, putting import ofUNetafter import ofmodel_factory, which is also importingUNet.UNetandLadderVAEmodules frommodels/__init__.py, mirrored changes in impacted scripts.UNetandLadderVAEmore explicit.Please ensure your PR meets the following requirements: