refactor(oscillator): #35 move to subfolder

kausalflow · Jun 15, 2024 · be219f5 · be219f5
1 parent b555f85
commit be219f5
Showing 1 changed file with 6 additions and 6 deletions.
diff --git a/hamilflow/models/harmonic_oscillator.py → ...ow/models/harmonic_oscillator/__init__.py b/hamilflow/models/harmonic_oscillator.py → ...ow/models/harmonic_oscillator/__init__.py
@@ -5,7 +5,7 @@
 import numpy as np
 import pandas as pd
 from numpy.typing import ArrayLike
-from pydantic import BaseModel, computed_field, field_validator
+from pydantic import BaseModel, Field, computed_field, field_validator
 
 
 class HarmonicOscillatorSystem(BaseModel):
@@ -57,14 +57,14 @@ def check_zeta_non_negative(cls, v: float) -> float:
 class HarmonicOscillatorIC(BaseModel):
  """The initial condition for a harmonic oscillator
 
- :cvar x0: the initial displacement
- :cvar v0: the initial velocity
+ :cvar x0: initial displacement
+ :cvar v0: initial velocity
  :cvar phi: initial phase
  """
 
- x0: float = 1.0
- v0: float = 0.0
- phi: float = 0.0
+ x0: float = Field(default=1.0)
+ v0: float = Field(default=0.0)
+ phi: float = Field(default=0.0)
 
 
 class HarmonicOscillatorBase(ABC):