diff --git a/launchable/commands/subset.py b/launchable/commands/subset.py index af2aa8ff7..0e2f9160f 100644 --- a/launchable/commands/subset.py +++ b/launchable/commands/subset.py @@ -49,6 +49,12 @@ help='subsetting by confidence from 0% to 100%', type=PERCENTAGE, ) +@click.option( + '--similarity', + 'similarity', + help='subsetting by similarity from 0 to 1.0. Tests scoring above this value are included in the subset.', + type=click.FloatRange(min=0, max=1.0), +) @click.option( '--goal-spec', 'goal_spec', @@ -242,6 +248,7 @@ def subset( test_suite: Optional[str] = None, is_get_tests_from_guess: bool = False, use_case: Optional[str] = None, + similarity: Optional[float] = None, ): app = context.obj tracking_client = TrackingClient(Command.SUBSET, app=app) @@ -510,6 +517,11 @@ def get_payload( "type": "subset-by-goal-spec", "goal": goal_spec } + elif similarity is not None: + payload["goal"] = { + "type": "subset-by-similarity", + "similarity": similarity + } else: payload['useServerSideOptimizationTarget'] = True