fix: add vicsek model in constructors

jung235 · Mar 6, 2024 · 52b298b · 52b298b
1 parent 557ee71
commit 52b298b
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 0 deletions.
diff --git 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
@@ -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",
 ]