From 52b298b4e71b355736e045056ea5e593ac84b659 Mon Sep 17 00:00:00 2001 From: jung235 Date: Wed, 6 Mar 2024 17:46:57 +0900 Subject: [PATCH] fix: add vicsek model in constructors --- pydiffuser/__init__.py | 4 ++++ pydiffuser/models/__init__.py | 3 +++ 2 files changed, 7 insertions(+) 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", ]