Skip to content

Commit 49930a0

Browse files
Fix pre-commit
1 parent 9e0b089 commit 49930a0

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

model_api/python/openvino/model_api/models/classification.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -185,7 +185,7 @@ def get_saliency_maps(self, outputs: Dict) -> np.ndarray:
185185
reordered_saliency_maps = [[] for _ in range(len(saliency_maps))]
186186
for batch in range(len(saliency_maps)):
187187
for label in self.labels:
188-
idx = self.hierarchical_info['cls_heads_info']['label_to_idx'][label]
188+
idx = self.hierarchical_info["cls_heads_info"]["label_to_idx"][label]
189189
reordered_saliency_maps[batch].append(saliency_maps[batch][idx])
190190
return np.array(reordered_saliency_maps)
191191

0 commit comments

Comments
 (0)