Merge: Constraint Refactor 3: Merge *NoLabelDuplicates* and *LinkedParameters* into *Repetition* (#881)

Closes #874
Based on #880

Author: Scienfitz

CHANGELOG.md

@@ -40,6 +40,9 @@ and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0
   `TransferLearningMode` enum) for selecting the kernel that models the task
   correlations in transfer learning, taking precedence over the task kernel of the
   configured kernel factory
+- `DiscreteRepetitionLimitConstraint` for controlling value repetition across parameters
+  via `n_max_repetitions` (replaces `DiscreteNoLabelDuplicatesConstraint` and
+  `DiscreteLinkedParametersConstraint`)
 
 ### Changed
 - `BOTORCH` GP preset now includes `BetaPrior(2.5, 1.5)` for the task covariance
@@ -63,6 +66,11 @@ and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0
 ### Deprecations
 - `DiscreteExcludeConstraint` in favor of
   `DiscreteSelectionConstraint(..., exclude=True)`
+- `DiscreteNoLabelDuplicatesConstraint` in favor of
+  `DiscreteRepetitionLimitConstraint(..., n_max_repetitions=1)`
+- `DiscreteLinkedParametersConstraint` in favor of
+  `DiscreteRepetitionLimitConstraint(..., n_max_repetitions=len(parameters)-1,
+  exclude=True)`
 
 ## [0.15.0] - 2026-06-11
 ### Breaking Changes

baybe/campaign.py

@@ -17,7 +17,7 @@
 from attrs.validators import instance_of
 from typing_extensions import override
 
-from baybe.constraints.base import DiscreteConstraint, DiscreteFilteringConstraint
+from baybe.constraints.base import DiscreteFilteringConstraint
 from baybe.exceptions import (
     IncompatibilityError,
     NoMeasurementsError,
@@ -426,7 +426,7 @@ def update_measurements(
 
     def toggle_discrete_candidates(  # noqa: DOC501
         self,
-        constraints: Collection[DiscreteConstraint] | pd.DataFrame,
+        constraints: Collection[DiscreteFilteringConstraint] | pd.DataFrame,
         exclude: bool,
         complement: bool = False,
         dry_run: bool = False,
@@ -436,8 +436,8 @@ def toggle_discrete_candidates(  # noqa: DOC501
         Args:
             constraints: A filtering mechanism determining the candidates subset to be
                 in-/excluded. Can be either a collection of
-                :class:`~baybe.constraints.base.DiscreteConstraint` or a dataframe.
-                For the latter, see :func:`~baybe.utils.dataframe.filter_df`
+                :class:`~baybe.constraints.base.DiscreteFilteringConstraint` or a
+                dataframe. For the latter, see :func:`~baybe.utils.dataframe.filter_df`
                 for details.
             exclude: If ``True``, the specified candidates are excluded.
                 If ``False``, the candidates are considered for recommendation.
@@ -483,8 +483,8 @@ def toggle_discrete_candidates(  # noqa: DOC501
 
         else:
             raise TypeError(
-                "Candidate toggling is not implemented for the given type of "
-                "constraint specifications."
+                f"Candidate toggling requires a dataframe or a collection of "
+                f"'{DiscreteFilteringConstraint.__name__}' instances."
             )
 
         if not dry_run:

baybe/constraints/__init__.py

@@ -16,6 +16,7 @@
     DiscreteNoLabelDuplicatesConstraint,
     DiscretePermutationInvarianceConstraint,
     DiscreteProductConstraint,
+    DiscreteRepetitionLimitConstraint,
     DiscreteSelectionConstraint,
     DiscreteSumConstraint,
 )
@@ -34,11 +35,12 @@
     "DiscreteCustomConstraint",
     "DiscreteDependenciesConstraint",
     "DiscreteExcludeConstraint",
-    "DiscreteSelectionConstraint",
     "DiscreteLinkedParametersConstraint",
     "DiscreteNoLabelDuplicatesConstraint",
     "DiscretePermutationInvarianceConstraint",
     "DiscreteProductConstraint",
+    "DiscreteRepetitionLimitConstraint",
+    "DiscreteSelectionConstraint",
     "DiscreteSumConstraint",
     # --- Other --- #
     "validate_constraints",

baybe/constraints/base.py

@@ -197,25 +197,31 @@ def has_polars_implementation(cls) -> bool:
             is not DiscreteFilteringConstraint._get_matching_rows_polars
         )
 
-    def get_invalid_polars(self) -> pl.Expr:
+    def get_invalid_polars(self, schema: pl.Schema) -> pl.Expr:
         """Translate the constraint to a Polars expression identifying rows to remove.
 
+        Args:
+            schema: The Polars schema of the dataframe being filtered.
+
         Returns:
             The Polars expression.
         """
-        matching_expr = self._get_matching_rows_polars()
+        matching_expr = self._get_matching_rows_polars(schema)
         if self.exclude:
             return matching_expr
         return ~matching_expr
 
-    def _get_matching_rows_polars(self) -> pl.Expr:
+    def _get_matching_rows_polars(self, schema: pl.Schema) -> pl.Expr:
         """Translate the constraint to a Polars expression identifying matching rows.
 
         Subclasses with a Polars implementation override this method. The expression
         should evaluate to ``True`` for rows that the specification matches/keeps
         (as if ``exclude=False``). The ``exclude`` inversion is applied by the base
         class in :meth:`get_invalid_polars`, not here.
 
+        Args:
+            schema: The Polars schema of the dataframe being filtered.
+
         Returns:
             A Polars expression that evaluates to ``True`` for matching rows.