diff --git a/model/src/pyrenew/latent/hospitaladmissions.py b/model/src/pyrenew/latent/hospitaladmissions.py index c6bc9028..359d293f 100644 --- a/model/src/pyrenew/latent/hospitaladmissions.py +++ b/model/src/pyrenew/latent/hospitaladmissions.py @@ -7,7 +7,7 @@ import numpyro.distributions as dist from numpy.typing import ArrayLike from pyrenew.distutil import validate_discrete_dist_vector -from pyrenew.metaclasses import RandomVariable +from pyrenew.metaclass import RandomVariable HospAdmissionsSample = namedtuple( "HospAdmissionsSample", diff --git a/model/src/pyrenew/latent/infections.py b/model/src/pyrenew/latent/infections.py index 409981de..19706e8c 100644 --- a/model/src/pyrenew/latent/infections.py +++ b/model/src/pyrenew/latent/infections.py @@ -11,7 +11,7 @@ reverse_discrete_dist_vector, validate_discrete_dist_vector, ) -from pyrenew.metaclasses import RandomVariable +from pyrenew.metaclass import RandomVariable InfectionsSample = namedtuple( "InfectionsSample", diff --git a/model/src/pyrenew/metaclasses.py b/model/src/pyrenew/metaclass.py similarity index 99% rename from model/src/pyrenew/metaclasses.py rename to model/src/pyrenew/metaclass.py index 7ea4e40f..96b337e0 100644 --- a/model/src/pyrenew/metaclasses.py +++ b/model/src/pyrenew/metaclass.py @@ -1,7 +1,7 @@ # -*- coding: utf-8 -*- """ -Observation helper classes +pyrenew helper classes """ from abc import ABCMeta, abstractmethod diff --git a/model/src/pyrenew/model/hospitalizations.py b/model/src/pyrenew/model/hospitalizations.py index 651c721d..26ea5609 100644 --- a/model/src/pyrenew/model/hospitalizations.py +++ b/model/src/pyrenew/model/hospitalizations.py @@ -2,7 +2,7 @@ from collections import namedtuple -from pyrenew.metaclasses import Model, RandomVariable, _assert_sample_and_rtype +from pyrenew.metaclass import Model, RandomVariable, _assert_sample_and_rtype from pyrenew.model.rtinfectionsrenewal import RtInfectionsRenewalModel from pyrenew.process import RtRandomWalkProcess diff --git a/model/src/pyrenew/model/rtinfectionsrenewal.py b/model/src/pyrenew/model/rtinfectionsrenewal.py index 3e132ee0..e302bf29 100644 --- a/model/src/pyrenew/model/rtinfectionsrenewal.py +++ b/model/src/pyrenew/model/rtinfectionsrenewal.py @@ -2,7 +2,7 @@ from collections import namedtuple -from pyrenew.metaclasses import Model, RandomVariable, _assert_sample_and_rtype +from pyrenew.metaclass import Model, RandomVariable, _assert_sample_and_rtype from pyrenew.process import RtRandomWalkProcess # Output class of the RtInfectionsRenewalModel diff --git a/model/src/pyrenew/observation/negativebinomial.py b/model/src/pyrenew/observation/negativebinomial.py index e5a7cff5..1aa416bc 100644 --- a/model/src/pyrenew/observation/negativebinomial.py +++ b/model/src/pyrenew/observation/negativebinomial.py @@ -5,7 +5,7 @@ import numpyro import numpyro.distributions as dist from numpy.typing import ArrayLike -from pyrenew.metaclasses import RandomVariable +from pyrenew.metaclass import RandomVariable class NegativeBinomialObservation(RandomVariable): diff --git a/model/src/pyrenew/observation/poisson.py b/model/src/pyrenew/observation/poisson.py index e687b9b6..f580eaec 100644 --- a/model/src/pyrenew/observation/poisson.py +++ b/model/src/pyrenew/observation/poisson.py @@ -2,7 +2,7 @@ import numpyro import numpyro.distributions as dist -from pyrenew.metaclasses import RandomVariable +from pyrenew.metaclass import RandomVariable class PoissonObservation(RandomVariable): diff --git a/model/src/pyrenew/process/ar.py b/model/src/pyrenew/process/ar.py index 8720945f..7d3d2969 100644 --- a/model/src/pyrenew/process/ar.py +++ b/model/src/pyrenew/process/ar.py @@ -5,7 +5,7 @@ import numpyro.distributions as dist from jax import lax from numpy.typing import ArrayLike -from pyrenew.metaclasses import RandomVariable +from pyrenew.metaclass import RandomVariable class ARProcess(RandomVariable): diff --git a/model/src/pyrenew/process/firstdifferencear.py b/model/src/pyrenew/process/firstdifferencear.py index afe31c32..5e71c027 100644 --- a/model/src/pyrenew/process/firstdifferencear.py +++ b/model/src/pyrenew/process/firstdifferencear.py @@ -2,7 +2,7 @@ import jax.numpy as jnp from numpy.typing import ArrayLike -from pyrenew.metaclasses import RandomVariable +from pyrenew.metaclass import RandomVariable from pyrenew.process import ARProcess diff --git a/model/src/pyrenew/process/rtrandomwalk.py b/model/src/pyrenew/process/rtrandomwalk.py index b0028335..55d452ba 100644 --- a/model/src/pyrenew/process/rtrandomwalk.py +++ b/model/src/pyrenew/process/rtrandomwalk.py @@ -2,7 +2,7 @@ import numpyro as npro import numpyro.distributions as dist -from pyrenew.metaclasses import RandomVariable +from pyrenew.metaclass import RandomVariable from pyrenew.process.simplerandomwalk import SimpleRandomWalkProcess from pyrenew.transform import AbstractTransform, LogTransform diff --git a/model/src/pyrenew/process/simplerandomwalk.py b/model/src/pyrenew/process/simplerandomwalk.py index c8227423..8f249c1d 100644 --- a/model/src/pyrenew/process/simplerandomwalk.py +++ b/model/src/pyrenew/process/simplerandomwalk.py @@ -3,7 +3,7 @@ import jax.numpy as jnp import numpyro as npro import numpyro.distributions as dist -from pyrenew.metaclasses import RandomVariable +from pyrenew.metaclass import RandomVariable class SimpleRandomWalkProcess(RandomVariable):