added __all__ to inits

AthenaEPI · Feb 2, 2018 · 1a085a7 · 1a085a7
1 parent d9baf72
commit 1a085a7
Show file tree

Hide file tree

Showing 7 changed files with 33 additions and 35 deletions.
diff --git a/dmipy/core/__init__.py b/dmipy/core/__init__.py
@@ -1,3 +1,7 @@
-#from __future__ import absolute_import, division, print_function
-#from .version import __version__  # noqa
-#from .acquisition_scheme import *  # noqa
+__all__ = [
+    "acquisition_scheme",
+    "constants",
+    "fitted_modeling_framework",
+    "gradient_conversions",
+    "modeling_framework"
+]
diff --git a/dmipy/data/__init__.py b/dmipy/data/__init__.py
@@ -0,0 +1,4 @@
+__all__ = [
+    "saved_acquisition_schemes",
+    "saved_data"
+]
diff --git a/dmipy/distributions/__init__.py b/dmipy/distributions/__init__.py
@@ -0,0 +1,4 @@
+__all__ = [
+    "distribute_models",
+    "distributions"
+]
diff --git a/dmipy/hcp_interface/__init__.py b/dmipy/hcp_interface/__init__.py
@@ -0,0 +1 @@
+__all__ = ["downloader_aws"]
diff --git a/dmipy/optimizers/__init__.py b/dmipy/optimizers/__init__.py
@@ -0,0 +1,4 @@
+__all__ = [
+    "brute2fine",
+    "mix"
+]
diff --git a/dmipy/signal_models/__init__.py b/dmipy/signal_models/__init__.py
@@ -1,29 +1,8 @@
-#"""Several tools for Diffusion MRI sequence simulation."""
-#from __future__ import division
-#from .constants import CONSTANTS
-#from .free_diffusion import free_diffusion_attenuation
-#from ..convert_units import (
-#    unit_conversion_factor,
-#    unit_conversion_factor_from_SI, unit_conversion_factor_to_SI
-#)
-#from .gradient_conversions import (
-#    q_from_g, g_from_q, b_from_g, g_from_b
-#)
-#
-#__all__ = [
-#    free_diffusion_attenuation,
-#    unit_conversion_factor, unit_conversion_factor_from_SI,
-#    unit_conversion_factor_to_SI,
-#    q_from_g, g_from_q, b_from_g, g_from_b
-#]
-#
-#
-#def max_diffusion_time(
-#    diameter=None,
-#    diffusion_constant=CONSTANTS['water_in_axons_diffusion_constant'],
-#):
-#    '''
-#    Diffusion time to reach container walls
-#    '''
-#    radius = diameter / 2
-#    return radius ** 2 / diffusion_constant
+__all__ = [
+    "capped_cylinder_models",
+    "cylinder_models",
+    "gaussian_models",
+    "plane_models",
+    "sphere_models",
+    "spherical_mean_models"
+]
diff --git a/dmipy/utils/__init__.py b/dmipy/utils/__init__.py
@@ -1,3 +1,5 @@
-#from __future__ import absolute_import, division, print_function
-#from .version import __version__  # noqa
-#from .acquisition_scheme import *  # noqa
+__all__ = [
+    "spherical_convolution",
+    "spherical_mean",
+    "utils"
+]