Merge pull request #22 from JuBiotech/fix-growth-import

Fix import compatibility

.github/workflows/pipeline.yml

@@ -11,7 +11,7 @@ jobs:
     strategy:
       matrix:
         python-version: [3.7, 3.8, 3.9]
-        pymc-version: ["without", "pymc==4.0.0b2", "pymc3"]
+        pymc-version: ["without", "pymc==4.0.0b6", "pymc3"]
     steps:
     - uses: actions/checkout@v2
     - name: Set up Python ${{ matrix.python-version }}

bletl/__init__.py

@@ -19,4 +19,4 @@
     NoMeasurementData,
 )
 
-__version__ = "1.1.0"
+__version__ = "1.1.1"

bletl/growth.py

@@ -5,7 +5,7 @@
 import calibr8
 import numpy
 import scipy.stats
-from calibr8.utils import at, pm
+from calibr8.utils import pm
 from packaging import version
 
 # Use the new ConstantData container if available,
@@ -16,6 +16,12 @@
     pmData = pm.Data
 
 
+try:
+    import aesara.tensor as at
+except ModuleNotFoundError:
+    import theano.tensor as at
+
+
 _log = logging.getLogger(__file__)
 
 
@@ -429,7 +435,7 @@ def fit_mu_t(
                 initval=mu_guess,
             )
 
-        X0 = pm.Lognormal("X0", mu=numpy.log(x0_prior), sd=1)
+        X0 = pm.LogNormal("X0", mu=numpy.log(x0_prior), sigma=1)
         Xt = pm.Deterministic(
             "X",
             at.concatenate([X0[None], X0 * pm.math.exp(at.extra_ops.cumsum(mu_t * dt))]),