remove ndims deprecation and extraneous code

Author: ColtAllen

pymc_extras/distributions/discrete.py

@@ -12,8 +12,6 @@
 #   See the License for the specific language governing permissions and
 #   limitations under the License.
 
-import warnings
-
 import numpy as np
 import pymc as pm
 
@@ -23,8 +21,6 @@
 from pytensor import tensor as pt
 from pytensor.tensor.random.op import RandomVariable
 
-warnings.filterwarnings("ignore", category=FutureWarning, message="ndims_params is deprecated")
-
 
 def log1mexp(x):
     cond = x < np.log(0.5)
@@ -409,18 +405,12 @@ def dist(cls, mu1, mu2, **kwargs):
 class GrassiaIIGeometricRV(RandomVariable):
     name = "g2g"
     signature = "(),(),()->()"
-    ndims_params = [0, 0, 0]  # deprecated in PyTensor 2.31.7, but still required for RandomVariable
 
     dtype = "int64"
     _print_name = ("GrassiaIIGeometric", "\\operatorname{GrassiaIIGeometric}")
 
     @classmethod
     def rng_fn(cls, rng, r, alpha, time_covariate_vector, size):
-        # Cast inputs as numpy arrays
-        r = np.asarray(r, dtype=np.float64)
-        alpha = np.asarray(alpha, dtype=np.float64)
-        time_covariate_vector = np.asarray(time_covariate_vector, dtype=np.float64)
-
         # Determine output size
         if size is None:
             size = np.broadcast_shapes(r.shape, alpha.shape, time_covariate_vector.shape)
@@ -525,8 +515,6 @@ def logp(value, r, alpha, time_covariate_vector=None):
         time_covariate_vector = pt.as_tensor_variable(time_covariate_vector)
 
         def C_t(t):
-            if t == 0:
-                return pt.constant(0.0)
             if time_covariate_vector.ndim == 0:
                 return t
             else:

tests/distributions/test_discrete.py

@@ -11,7 +11,6 @@
 #   WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 #   See the License for the specific language governing permissions and
 #   limitations under the License.
-import warnings
 
 import numpy as np
 import pymc as pm
@@ -20,8 +19,6 @@
 import pytest
 import scipy.stats
 
-warnings.filterwarnings("ignore", category=FutureWarning, message="ndims_params is deprecated")
-
 from pymc.testing import (
     BaseTestDistributionRandom,
     Domain,