Consistent decimal scale in polars

Author: cgevans

src/alhambra_mixes/actions.py

@@ -291,7 +291,7 @@ def all_components_polars(
             r = _ratio(dc, sc)
             if math.isnan(r):
                 r = None
-            comps = comps.with_columns(pl.col("concentration_nM").mul(r))
+            comps = comps.with_columns(pl.col("concentration_nM").mul(r).cast(pl.Decimal(scale=6)))
 
             all_comps.append(comps)
 

src/alhambra_mixes/components.py

@@ -186,7 +186,7 @@ def all_components_polars(self, _cache_key=None) -> pl.DataFrame:
             },
             schema={
                 'name': pl.String,
-                'concentration_nM': pl.Decimal(precision=20, scale=6),
+                'concentration_nM': pl.Decimal(scale=6),
                 'component': pl.Object
             }
         )

src/alhambra_mixes/mixes.py

@@ -522,7 +522,7 @@ def all_components_polars(self, _cache_key=None) -> pl.DataFrame:
             pl.when(pl.col("concentration_nM").is_null().any())
             .then(pl.lit(None))
             .otherwise(pl.col("concentration_nM").sum())
-            .alias("concentration_nM"),
+            .alias("concentration_nM").cast(pl.Decimal(scale=6)),
             pl.col("component").first(),  # FIXME
         )