fix

nifty5/__init__.py

@@ -80,6 +80,8 @@ from .library.los_response import LOSResponse
 from .library.wiener_filter_curvature import WienerFilterCurvature
 from .library.correlated_fields import CorrelatedField, MfCorrelatedField
 
+from . import extra
+
 from .utilities import memo, frozendict
 
 from .logger import logger

nifty5/extra.py

@@ -20,10 +20,10 @@ from __future__ import absolute_import, division, print_function
 
 import numpy as np
 
-from ..compat import *
-from ..field import Field
-from ..linearization import Linearization
-from ..sugar import from_random
+from .compat import *
+from .field import Field
+from .linearization import Linearization
+from .sugar import from_random
 
 __all__ = ["consistency_check", "check_value_gradient_consistency",
            "check_value_gradient_metric_consistency"]

nifty5/plot.py

@@ -22,13 +22,13 @@ import os
 
 import numpy as np
 
-from .. import dobj
-from ..compat import *
-from ..domains.gl_space import GLSpace
-from ..domains.hp_space import HPSpace
-from ..domains.power_space import PowerSpace
-from ..domains.rg_space import RGSpace
-from ..field import Field
+from . import dobj
+from .compat import *
+from .domains.gl_space import GLSpace
+from .domains.hp_space import HPSpace
+from .domains.power_space import PowerSpace
+from .domains.rg_space import RGSpace
+from .field import Field
 
 # relevant properties:
 # - x/y size