diff --git a/src/matchbox/client/results.py b/src/matchbox/client/results.py index 95de99dd..b4d38f89 100644 --- a/src/matchbox/client/results.py +++ b/src/matchbox/client/results.py @@ -127,7 +127,7 @@ def _check_range(arr: pa.Array, min: float, max: float) -> bool: target_type=pa.uint8(), allow_float_truncate=True ), ) - value.set_column( + value = value.set_column( i=value.schema.get_field_index("probability"), field_="probability", column=probability_uint8, diff --git a/src/matchbox/server/postgresql/orm.py b/src/matchbox/server/postgresql/orm.py index dc2dc291..a7725e0b 100644 --- a/src/matchbox/server/postgresql/orm.py +++ b/src/matchbox/server/postgresql/orm.py @@ -285,5 +285,5 @@ class Probabilities(CountMixin, MBDB.MatchboxBase): # Constraints __table_args__ = ( - CheckConstraint("probability BETWEEN 0 AND 1", name="valid_probability"), + CheckConstraint("probability BETWEEN 0 AND 100", name="valid_probability"), )