Suppressions for pyrefly

PiperOrigin-RevId: 951112155

Author: h-joo

lit_nlp/app.py

@@ -305,7 +305,7 @@ def _get_preds(self,
       t_class = getattr(types, t_name, None)
       if not issubclass(t_class, types.LitType):  # pyrefly: ignore[bad-argument-type]
         raise TypeError(f"Class '{t_name}' is not a valid LitType.")
-      requested_fields.extend(utils.find_spec_keys(output_spec, t_class))
+      requested_fields.extend(utils.find_spec_keys(output_spec, t_class))  # pyrefly: ignore[missing-attribute]
     ret_keys = set(requested_fields)  # de-dupe  # pyrefly: ignore[bad-argument-type]
 
     # Return selected keys.

lit_nlp/components/index.py

@@ -63,8 +63,8 @@ def __init__(
     # Indicator whether to build new indices. If False, only load existing ones.
     self._initialize_new_indices = initialize_new_indices
     # Ensure directory to save indices exists.
-    if not os.path.isdir(data_dir):
-      os.mkdir(data_dir)
+    if not os.path.isdir(data_dir):  # pyrefly: ignore[bad-argument-type]
+      os.mkdir(data_dir)  # pyrefly: ignore[bad-argument-type]
     self._data_dir = data_dir
     self._models = models
 
@@ -93,12 +93,12 @@ def _get_lookup_key(self, model_name, dataset_name):
 
   def _get_index_path(self, index_key):
     """Get the file path for an index."""
-    file_path = os.path.join(self._data_dir, f"{index_key}.ann")
+    file_path = os.path.join(self._data_dir, f"{index_key}.ann")  # pyrefly: ignore[no-matching-overload]
     return file_path
 
   def _get_lookup_path(self, lookup_key):
     """Get the file path for the lookup index."""
-    file_path = os.path.join(self._data_dir, lookup_key + ".pkl")
+    file_path = os.path.join(self._data_dir, lookup_key + ".pkl")  # pyrefly: ignore[no-matching-overload]
     return file_path
 
   def _create_empty_indices(self, model_name, dataset_name):

lit_nlp/components/thresholder.py

@@ -238,7 +238,7 @@ def run(
           list(inputs), model, dataset, model_outputs, metrics_config))
 
       # Get and store the metrics for each facet of the dataset for this margin.
-      if not (facets := config.facets):
+      if not (facets := config.facets):  # pyrefly: ignore[missing-attribute]
         continue
 
       for facet_key, facet_dict in facets.items():