Skip to content

Commit 46d18b4

Browse files
committed
Deprecate Model.cont_vars and Model.disc_vars in favor of more explicit names
1 parent 4511792 commit 46d18b4

File tree

5 files changed

+31
-15
lines changed

5 files changed

+31
-15
lines changed

pymc/model.py

Lines changed: 26 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -975,6 +975,32 @@ def unobserved_value_vars(self):
975975

976976
return vars + untransformed_vars + deterministics
977977

978+
@property
979+
def disc_vars(self):
980+
warnings.warn(
981+
"Model.disc_vars has been deprecated. Use Model.discrete_value_vars instead.",
982+
FutureWarning,
983+
)
984+
return self.discrete_value_vars
985+
986+
@property
987+
def discrete_value_vars(self):
988+
"""All the discrete value variables in the model"""
989+
return list(typefilter(self.value_vars, discrete_types))
990+
991+
@property
992+
def cont_vars(self):
993+
warnings.warn(
994+
"Model.cont_vars has been deprecated. Use Model.continuous_value_vars instead.",
995+
FutureWarning,
996+
)
997+
return self.continuous_value_vars
998+
999+
@property
1000+
def continuous_value_vars(self):
1001+
"""All the continuous value variables in the model"""
1002+
return list(typefilter(self.value_vars, continuous_types))
1003+
9781004
@property
9791005
def basic_RVs(self):
9801006
"""List of random variables the model is defined in terms of
@@ -1019,16 +1045,6 @@ def unobserved_RVs(self):
10191045
"""
10201046
return self.free_RVs + self.deterministics
10211047

1022-
@property
1023-
def disc_vars(self):
1024-
"""All the discrete variables in the model"""
1025-
return list(typefilter(self.value_vars, discrete_types))
1026-
1027-
@property
1028-
def cont_vars(self):
1029-
"""All the continuous variables in the model"""
1030-
return list(typefilter(self.value_vars, continuous_types))
1031-
10321048
@property
10331049
def test_point(self) -> Dict[str, np.ndarray]:
10341050
"""Deprecated alias for `Model.initial_point(seed=None)`."""

pymc/step_methods/hmc/base_hmc.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -88,7 +88,7 @@ def __init__(
8888
self._model = modelcontext(model)
8989

9090
if vars is None:
91-
vars = self._model.cont_vars
91+
vars = self._model.continuous_value_vars
9292
else:
9393
vars = [self._model.rvs_to_values.get(var, var) for var in vars]
9494

pymc/step_methods/metropolis.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -773,7 +773,7 @@ def __init__(
773773
initial_values_size = sum(initial_values[n.name].size for n in model.value_vars)
774774

775775
if vars is None:
776-
vars = model.cont_vars
776+
vars = model.continuous_value_vars
777777
else:
778778
vars = [model.rvs_to_values.get(var, var) for var in vars]
779779
vars = pm.inputvars(vars)
@@ -924,7 +924,7 @@ def __init__(
924924
initial_values_size = sum(initial_values[n.name].size for n in model.value_vars)
925925

926926
if vars is None:
927-
vars = model.cont_vars
927+
vars = model.continuous_value_vars
928928
else:
929929
vars = [model.rvs_to_values.get(var, var) for var in vars]
930930
vars = pm.inputvars(vars)

pymc/step_methods/slicer.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -63,7 +63,7 @@ def __init__(self, vars=None, w=1.0, tune=True, model=None, iter_limit=np.inf, *
6363
self.iter_limit = iter_limit
6464

6565
if vars is None:
66-
vars = self.model.cont_vars
66+
vars = self.model.continuous_value_vars
6767
else:
6868
vars = [self.model.rvs_to_values.get(var, var) for var in vars]
6969
vars = inputvars(vars)

pymc/tuning/starting.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -92,7 +92,7 @@ def find_MAP(
9292
model = modelcontext(model)
9393

9494
if vars is None:
95-
vars = model.cont_vars
95+
vars = model.continuous_value_vars
9696
if not vars:
9797
raise ValueError("Model has no unobserved continuous variables.")
9898
else:

0 commit comments

Comments
 (0)