fix: round kev energy

2026-08-20 14:46:27 +02:00
parent 47bfe56cad
commit b2992471ce
1 changed files with 6 additions and 2 deletions
@@ -1,11 +1,15 @@
 import math
 from typing import Annotated

-from pydantic import BaseModel, Field
+from pydantic import AfterValidator, BaseModel, Field
+
+
+def round_dp(ndigits: int, /) -> AfterValidator:
+    return AfterValidator(lambda v: round(v, ndigits))


 class DiffractionGeometry(BaseModel):
-    energy_keV: Annotated[float, Field(gt=1.0, lt=100.0)]
+    energy_keV: Annotated[float, Field(gt=1.0, lt=100.0), round_dp(3)]
    dtz_mm: Annotated[float, Field(gt=10.0, lt=5000.0)]
    pixel_size_mm: Annotated[float, Field(ge=0.05, le=0.5)]
    beam_center_pxl: tuple[float, float]