Replace data fetch with lookup (#4788)

Summary:

Instead of fetching data we should use lookup for performance reasons.

Reviewed By: saitcakmak

Differential Revision: D91011958

Author: mgarrard

ax/generation_strategy/transition_criterion.py

@@ -581,9 +581,8 @@ def is_met(
         experiment: Experiment,
         curr_node: GenerationNode,
     ) -> bool:
-        # TODO: @mgarrard replace fetch_data with lookup_data
-        data = experiment.fetch_data(
-            metrics=[experiment.signature_to_metric[self.metric_signature]]
+        data = experiment.lookup_data().filter(
+            metric_signatures=[self.metric_signature]
         )
 
         count_no = (data.df["mean"] == 0).sum()