diff --git a/nifty/energies/wiener_filter_energy.py b/nifty/energies/wiener_filter_energy.py index 52714fe221a65f88676c9b664b9fb584a351e302..ed2d0d57dc168866bea65923d8dd78056b727080 100644 --- a/nifty/energies/wiener_filter_energy.py +++ b/nifty/energies/wiener_filter_energy.py @@ -1,5 +1,5 @@ from .energy import Energy -from nifty.operators.curvature_operators import WienerFilterCurvature +from nifty.operators import WienerFilterCurvature class WienerFilterEnergy(Energy): """The Energy for the Wiener filter. diff --git a/nifty/operators/__init__.py b/nifty/operators/__init__.py index 6e066d944672d363d047daebf0c360164a59cf58..b64ee506ada2e52aa5c5397b4e0cc5f23f3c7f13 100644 --- a/nifty/operators/__init__.py +++ b/nifty/operators/__init__.py @@ -40,4 +40,4 @@ from composed_operator import ComposedOperator from response_operator import ResponseOperator -from curvature_operators import * +from curvature_operators import WienerFilterCurvature