Rename _get_default_transform to _default_transform

Author: ricardoV94

pymc/distributions/continuous.py

@@ -87,7 +87,7 @@ def polyagamma_cdf(*args, **kwargs):
 )
 from pymc.distributions.distribution import DIST_PARAMETER_TYPES, Continuous
 from pymc.distributions.shape_utils import rv_size_is_none
-from pymc.distributions.transforms import _get_default_transform
+from pymc.distributions.transforms import _default_transform
 from pymc.math import invlogit, logdiffexp, logit
 from pymc.util import UNSET
 
@@ -140,17 +140,17 @@ class CircularContinuous(Continuous):
     """Base class for circular continuous distributions"""
 
 
-@_get_default_transform.register(PositiveContinuous)
+@_default_transform.register(PositiveContinuous)
 def pos_cont_transform(op):
     return transforms.log
 
 
-@_get_default_transform.register(UnitContinuous)
+@_default_transform.register(UnitContinuous)
 def unit_cont_transform(op):
     return transforms.logodds
 
 
-@_get_default_transform.register(CircularContinuous)
+@_default_transform.register(CircularContinuous)
 def circ_cont_transform(op):
     return transforms.circular
 

pymc/distributions/transforms.py

@@ -42,7 +42,7 @@
 
 
 @singledispatch
-def _get_default_transform(op: Op):
+def _default_transform(op: Op):
     """Return default transform for a given Distribution `Op`"""
     return None
 

pymc/model.py

@@ -60,7 +60,7 @@
 from pymc.data import GenTensorVariable, Minibatch
 from pymc.distributions import joint_logpt
 from pymc.distributions.logprob import _get_scaling
-from pymc.distributions.transforms import _get_default_transform
+from pymc.distributions.transforms import _default_transform
 from pymc.exceptions import ImputationWarning, SamplingError, ShapeError
 from pymc.initial_point import make_initial_point_fn
 from pymc.math import flatten_list
@@ -1422,7 +1422,7 @@ def create_value_var(
         # Make the value variable a transformed value variable,
         # if there's an applicable transform
         if transform is UNSET and rv_var.owner:
-            transform = _get_default_transform(rv_var.owner.op)
+            transform = _default_transform(rv_var.owner.op)
 
         if transform is not None and transform is not UNSET:
             value_var.tag.transform = transform