Internal

PiperOrigin-RevId: 551858134
google-research · Jul 28, 2023 · 0bc69f8 · 0bc69f8
1 parent 5f9285c
commit 0bc69f8
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/chirp/eval/eval_lib.py b/chirp/eval/eval_lib.py
@@ -464,7 +464,7 @@ def _prepare_eval_set(
   for class_name in eval_set_specification.class_names:
     choice_key, rng_key = jax.random.split(rng_key)
 
-    class_representative_mask = global_class_representative_mask & _df_eval(  # pytype: disable=unsupported-operands  # typed-pandas
+    class_representative_mask = global_class_representative_mask & _df_eval(
         embeddings_df,
         eval_set_specification.class_representative_classwise_mask_fn(
             class_name
@@ -490,7 +490,7 @@ def _prepare_eval_set(
     )
 
     search_corpus_mask = (
-        global_search_corpus_mask  # pytype: disable=unsupported-operands  # typed-pandas
+        global_search_corpus_mask
         & _df_eval(
             embeddings_df,
             eval_set_specification.search_corpus_classwise_mask_fn(class_name),