refactor(oscillator): #35 factor out initial conditions

hamilflow/models/harmonic_oscillator/__init__.py

@@ -5,7 +5,9 @@
 import numpy as np
 import pandas as pd
 from numpy.typing import ArrayLike
-from pydantic import BaseModel, Field, computed_field, field_validator
+from pydantic import BaseModel, computed_field, field_validator
+
+from hamilflow.models.harmonic_oscillator.initial_conditions import HarmonicOscillatorIC
 
 
 class HarmonicOscillatorSystem(BaseModel):
@@ -54,19 +56,6 @@ def check_zeta_non_negative(cls, v: float) -> float:
         return v
 
 
-class HarmonicOscillatorIC(BaseModel):
-    """The initial condition for a harmonic oscillator
-
-    :cvar x0: initial displacement
-    :cvar v0: initial velocity
-    :cvar phi: initial phase
-    """
-
-    x0: float = Field(default=1.0)
-    v0: float = Field(default=0.0)
-    phi: float = Field(default=0.0)
-
-
 class HarmonicOscillatorBase(ABC):
     r"""Base class to generate time series data
     for a [harmonic oscillator](https://en.wikipedia.org/wiki/Harmonic_oscillator).

hamilflow/models/harmonic_oscillator/initial_conditions.py

@@ -0,0 +1,14 @@
+from pydantic import BaseModel, Field
+
+
+class HarmonicOscillatorIC(BaseModel):
+    """The initial condition for a harmonic oscillator
+
+    :cvar x0: initial displacement
+    :cvar v0: initial velocity
+    :cvar phi: initial phase
+    """
+
+    x0: float = Field(default=1.0)
+    v0: float = Field(default=0.0)
+    phi: float = Field(default=0.0)