[AIENG-312] Add a new optimization target option: --similarity

Author: ono-max

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