transforms in mcmc/api.py
@@ -384,7 +384,8 @@ class MCMC(object):
# If transforms is not explicitly provided, infer automatically using
# model args, kwargs.
- if self.transforms is None and isinstance(self.kernel, (HMC, NUTS)):
+ if self.transforms is None:
+ if hasattr(self.kernel, 'transforms'):
if self.kernel.transforms is not None:
self.transforms = self.kernel.transforms
elif self.kernel.model:
