fix(browniand_motion): Fix typing of BrownianMotion x0 validator

kausalflow · Mar 17, 2024 · 961593f · 961593f
1 parent 682cee9
commit 961593f
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/hamilflow/models/brownian_motion.py b/hamilflow/models/brownian_motion.py
@@ -58,7 +58,7 @@ class BrownianMotionIC(BaseModel):
 
     @field_validator("x0")
     @classmethod
-    def check_x0_types(cls, v: Union[float, int, list]) -> float:
+    def check_x0_types(cls, v: Union[float, int, list]) -> np.ndarray:
         if not isinstance(v, (float, int, list)):
             raise ValueError(f"Value of x0 should be int/float/list of int/float: {v=}")