diff --git a/pydiffuser/__init__.py b/pydiffuser/__init__.py index 8ec01c8..9c4f696 100644 --- a/pydiffuser/__init__.py +++ b/pydiffuser/__init__.py @@ -12,6 +12,8 @@ RunAndTumbleParticleConfig, SmoluchowskiEquation, SmoluchowskiEquationConfig, + VicsekModel, + VicsekModelConfig, ) from .tracer import Ensemble, Trajectory from .utils import load, save @@ -30,6 +32,8 @@ "RunAndTumbleParticleConfig", "SmoluchowskiEquation", "SmoluchowskiEquationConfig", + "VicsekModel", + "VicsekModelConfig", "Ensemble", "Trajectory", "load", diff --git a/pydiffuser/models/__init__.py b/pydiffuser/models/__init__.py index 7d45dd5..2dcbf8a 100644 --- a/pydiffuser/models/__init__.py +++ b/pydiffuser/models/__init__.py @@ -9,6 +9,7 @@ from .levy import LevyWalk, LevyWalkConfig from .rtp import RunAndTumbleParticle, RunAndTumbleParticleConfig from .smoluchowski import SmoluchowskiEquation, SmoluchowskiEquationConfig +from .vicsek import VicsekModel, VicsekModelConfig __all__ = [ "ActiveBrownianParticle", @@ -23,6 +24,8 @@ "RunAndTumbleParticleConfig", "SmoluchowskiEquation", "SmoluchowskiEquationConfig", + "VicsekModel", + "VicsekModelConfig", ]