Skip to content

Commit a32d30c

Browse files
h-jooLIT team
authored andcommitted
Automated Code Change
PiperOrigin-RevId: 752850543
1 parent 3339dc6 commit a32d30c

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

lit_nlp/lib/caching.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -307,7 +307,7 @@ def predict(self,
307307
len(miss_idxs), len(cached_results))
308308
else:
309309
# If all results were already cached, return them.
310-
return cached_results
310+
return cached_results # pytype: disable=bad-return-type
311311

312312
with self._cache.get_pred_lock(input_keys):
313313
model_preds = list(self.wrapped.predict(progress_indicator(misses)))
@@ -326,7 +326,7 @@ def predict(self,
326326
# Remove the prediction lock from the cache as the request is complete
327327
self._cache.delete_pred_lock(input_keys)
328328

329-
return cached_results
329+
return cached_results # pytype: disable=bad-return-type
330330

331331
def _get_results_from_cache(self, input_keys: list[CacheKey]):
332332
with self._cache.lock:

0 commit comments

Comments
 (0)