Skip to content

Commit df627ea

Browse files
Chris Fonnesbecktwiecki
authored andcommitted
Remove import * statements (#1257)
* Removed import * statements from most files * Additional import fixes * Additional import fixes * Fix for relative import of sampling * Most tests pass * Changed relative to absolute import for sampling within ATMCMC * Fixed circular import issue * Fixed import syntax error * Fixes for additional import errors * MAINT Add failing imports. * MAINT Add failing imports. * More namespace fixes for tests
1 parent 0222c85 commit df627ea

Some content is hidden

Large Commits have some content hidden by default. Use the searchbox below for content that may be hidden.

41 files changed

+397
-234
lines changed

pymc3/__init__.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,15 +1,15 @@
11
__version__ = "3.0"
22

3-
from .core import *
43
from .distributions import *
54
from .math import *
6-
5+
from .model import *
76
from .stats import *
87
from .sampling import *
98
from .interactive_sampling import *
109
from .step_methods import *
1110
from .tuning import *
1211
from .variational import *
12+
from . import sampling
1313

1414
from .debug import *
1515

pymc3/core.py

Lines changed: 0 additions & 5 deletions
This file was deleted.

pymc3/debug.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,5 @@
11
import numpy as np
2-
from .blocking import *
2+
from .blocking import DictToVarBijection
33

44
# TODO I could not locate this function used anywhere in the code base
55
# do we need it?

pymc3/distributions/__init__.py

Lines changed: 109 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -1,8 +1,112 @@
11
from . import timeseries
22
from . import transforms
33

4-
from .continuous import *
5-
from .discrete import *
6-
from .distribution import *
7-
from .multivariate import *
8-
from .timeseries import *
4+
from .continuous import Uniform
5+
from .continuous import Flat
6+
from .continuous import Normal
7+
from .continuous import Beta
8+
from .continuous import Exponential
9+
from .continuous import Laplace
10+
from .continuous import StudentT
11+
from .continuous import Cauchy
12+
from .continuous import HalfCauchy
13+
from .continuous import Gamma
14+
from .continuous import Weibull
15+
from .continuous import Bound
16+
from .continuous import StudentTpos
17+
from .continuous import Lognormal
18+
from .continuous import ChiSquared
19+
from .continuous import HalfNormal
20+
from .continuous import Wald
21+
from .continuous import Pareto
22+
from .continuous import InverseGamma
23+
from .continuous import ExGaussian
24+
from .continuous import VonMises
25+
26+
from .discrete import Binomial
27+
from .discrete import BetaBinomial
28+
from .discrete import Bernoulli
29+
from .discrete import Poisson
30+
from .discrete import NegativeBinomial
31+
from .discrete import ConstantDist
32+
from .discrete import ZeroInflatedPoisson
33+
from .discrete import DiscreteUniform
34+
from .discrete import Geometric
35+
from .discrete import Categorical
36+
37+
from .distribution import DensityDist
38+
from .distribution import Distribution
39+
from .distribution import Continuous
40+
from .distribution import Discrete
41+
from .distribution import NoDistribution
42+
from .distribution import TensorType
43+
from .distribution import draw_values
44+
45+
from .multivariate import MvNormal
46+
from .multivariate import MvStudentT
47+
from .multivariate import Dirichlet
48+
from .multivariate import Multinomial
49+
from .multivariate import Wishart
50+
from .multivariate import WishartBartlett
51+
from .multivariate import LKJCorr
52+
53+
from .timeseries import AR1
54+
from .timeseries import GaussianRandomWalk
55+
from .timeseries import GARCH11
56+
57+
from .transforms import transform
58+
from .transforms import stick_breaking
59+
from .transforms import logodds
60+
from .transforms import log
61+
from .transforms import sum_to_1
62+
63+
__all__ = ['Uniform',
64+
'Flat',
65+
'Normal',
66+
'Beta',
67+
'Exponential',
68+
'Laplace',
69+
'StudentT',
70+
'Cauchy',
71+
'HalfCauchy',
72+
'Gamma',
73+
'Weibull',
74+
'Bound',
75+
'StudentTpos',
76+
'Lognormal',
77+
'ChiSquared',
78+
'HalfNormal',
79+
'Wald',
80+
'Pareto',
81+
'InverseGamma',
82+
'ExGaussian',
83+
'VonMises',
84+
'Binomial',
85+
'BetaBinomial',
86+
'Bernoulli',
87+
'Poisson',
88+
'NegativeBinomial',
89+
'ConstantDist',
90+
'ZeroInflatedPoisson',
91+
'DiscreteUniform',
92+
'Geometric',
93+
'Categorical',
94+
'DensityDist',
95+
'Distribution',
96+
'Continuous',
97+
'Discrete',
98+
'NoDistribution',
99+
'TensorType',
100+
'MvNormal',
101+
'MvStudentT',
102+
'Dirichlet',
103+
'Multinomial',
104+
'Wishart',
105+
'WishartBartlett',
106+
'LKJCorr',
107+
'AR1',
108+
'GaussianRandomWalk',
109+
'GARCH11'
110+
]
111+
112+

pymc3/distributions/distribution.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -6,7 +6,7 @@
66
from ..model import Model, get_named_nodes
77

88

9-
__all__ = ['DensityDist', 'Distribution', 'Continuous', 'Discrete', 'NoDistribution', 'TensorType']
9+
__all__ = ['DensityDist', 'Distribution', 'Continuous', 'Discrete', 'NoDistribution', 'TensorType', 'draw_values']
1010

1111

1212
class Distribution(object):

pymc3/distributions/transforms.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -6,7 +6,7 @@
66
from ..math import logit
77
import numpy as np
88

9-
__all__ = ['transform', 'stick_breaking', 'logodds', 'log']
9+
__all__ = ['transform', 'stick_breaking', 'logodds', 'log', 'sum_to_1']
1010

1111

1212
class Transform(object):

pymc3/examples/ARM12_6uranium.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -55,8 +55,8 @@ def run(n=3000):
5555

5656
start = Point({
5757
'groupmean': obs_means.mean(),
58-
'groupsd_interval': 0,
59-
'sd_interval': 0,
58+
'groupsd_interval_': 0,
59+
'sd_interval_': 0,
6060
'means': np.array(obs_means),
6161
'u_m': np.array([.72]),
6262
'floor_m': 0.,

pymc3/examples/ATMIP_2gaussians.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,6 @@
11
import pymc3 as pm
22
import numpy as np
3-
from pymc3.step_methods import ATMCMC as atmcmc
3+
from ..step_methods import ATMCMC, ATMIP_sample
44
import theano.tensor as tt
55
from matplotlib import pylab as plt
66

@@ -28,7 +28,7 @@
2828
def two_gaussians(x):
2929
log_like1 = - 0.5 * n * tt.log(2 * np.pi) \
3030
- 0.5 * tt.log(dsigma) \
31-
- 0.5 * (x - mu1).tt.dot(isigma).dot(x - mu1)
31+
- 0.5 * (x - mu1).T.dot(isigma).dot(x - mu1)
3232
log_like2 = - 0.5 * n * tt.log(2 * np.pi) \
3333
- 0.5 * tt.log(dsigma) \
3434
- 0.5 * (x - mu2).T.dot(isigma).dot(x - mu2)
@@ -45,10 +45,10 @@ def two_gaussians(x):
4545
llk = pm.Potential('like', like)
4646

4747
with ATMIP_test:
48-
step = atmcmc.ATMCMC(n_chains=n_chains, tune_interval=tune_interval,
48+
step = ATMCMC(n_chains=n_chains, tune_interval=tune_interval,
4949
likelihood_name=ATMIP_test.deterministics[0].name)
5050

51-
trcs = atmcmc.ATMIP_sample(
51+
trcs = ATMIP_sample(
5252
n_steps=n_steps,
5353
step=step,
5454
njobs=njobs,

pymc3/examples/GHME_2013.py

Lines changed: 3 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -62,7 +62,7 @@ def interpolate(x0,y0, x, group):
6262
with Model() as model:
6363
coeff_sd = StudentT('coeff_sd', 10, 1, 5**-2)
6464

65-
y = GaussianRandomWalk('y', sd=coeff_sd, shape = (nknots, ncountries))
65+
y = GaussianRandomWalk('y', sd=coeff_sd, shape=(nknots, ncountries))
6666

6767
p = interpolate(knots, y, age, group)
6868

@@ -78,14 +78,14 @@ def interpolate(x0,y0, x, group):
7878
# <codecell>
7979

8080
with model:
81-
s = find_MAP( vars=[sd, y])
81+
s = find_MAP(vars=[sd, y])
8282

8383
step = NUTS(scaling = s)
8484
trace = sample(100, step, s)
8585

8686
s = trace[-1]
8787

88-
step = NUTS(scaling = s)
88+
step = NUTS(scaling=s)
8989

9090
def run(n=3000):
9191
if n == "short":
@@ -106,13 +106,6 @@ def run(n=3000):
106106

107107
ylim(0,rate.max())
108108

109-
# <codecell>
110-
111-
traceplot(trace[100:], varnames = [coeff_sd,sd ]);
112-
113-
# <codecell>
114-
115-
autocorrplot(trace, varnames = [coeff_sd,sd ])
116109

117110
if __name__ == '__main__':
118111
run()

pymc3/examples/glm_linear.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,7 +1,7 @@
11
import numpy as np
22
import scipy.optimize as opt
33

4-
from pymc3 import *
4+
from pymc3 import Model, glm, find_MAP, sample
55

66
# Generate data
77
size = 50

0 commit comments

Comments
 (0)