File tree Expand file tree Collapse file tree 1 file changed +2
-2
lines changed Expand file tree Collapse file tree 1 file changed +2
-2
lines changed Original file line number Diff line number Diff line change @@ -307,7 +307,7 @@ def predict(self,
307
307
len (miss_idxs ), len (cached_results ))
308
308
else :
309
309
# If all results were already cached, return them.
310
- return cached_results
310
+ return cached_results # pytype: disable=bad-return-type
311
311
312
312
with self ._cache .get_pred_lock (input_keys ):
313
313
model_preds = list (self .wrapped .predict (progress_indicator (misses )))
@@ -326,7 +326,7 @@ def predict(self,
326
326
# Remove the prediction lock from the cache as the request is complete
327
327
self ._cache .delete_pred_lock (input_keys )
328
328
329
- return cached_results
329
+ return cached_results # pytype: disable=bad-return-type
330
330
331
331
def _get_results_from_cache (self , input_keys : list [CacheKey ]):
332
332
with self ._cache .lock :
You can’t perform that action at this time.
0 commit comments