include all other modules

better · May 30, 2018 · a142291 · a142291
1 parent e1d7401
commit a142291
Show file tree

Hide file tree

Showing 5 changed files with 20 additions and 0 deletions.
diff --git a/convoys/multi.py b/convoys/multi.py
@@ -2,6 +2,8 @@
 from convoys import regression
 from convoys import single
 
+__all__ = ['KaplanMeier', 'Exponential', 'Weibull', 'Gamma', 'GeneralizedGamma']
+
 
 class MultiModel:
     pass  # TODO

diff --git a/convoys/plotting.py b/convoys/plotting.py
@@ -2,6 +2,8 @@
 from matplotlib import pyplot
 import convoys.multi
 
+__all__ = ['plot_cohorts']
+
 
 _models = {
     'kaplan-meier': lambda ci: convoys.multi.KaplanMeier(),

diff --git a/convoys/single.py b/convoys/single.py
@@ -2,6 +2,8 @@
 from scipy.special import expit, logit
 import scipy.stats
 
+__all__ = ['KaplanMeier']
+
 
 class SingleModel:
     pass  # TODO

diff --git a/convoys/utils.py b/convoys/utils.py
@@ -1,6 +1,8 @@
 import datetime
 import pandas
 
+__all__ = ['get_arrays']
+
 
 def get_timescale(t):
     ''' Take a datetime or a numerical type, return two things:

diff --git a/docs/index.rst b/docs/index.rst
@@ -1,8 +1,20 @@
 Full API documentation
 ======================
 
+.. automodule:: convoys.plotting
+   :members:
+
 .. automodule:: convoys.regression
    :members:
 
+.. automodule:: convoys.single
+   :members:
+
+.. automodule:: convoys.multi
+   :members:
+
+.. automodule:: convoys.utils
+   :members:
+
 .. automodule:: convoys.gamma
    :members: