Skip to content

Commit b38bb3e

Browse files
authored
Address pylint pytorch_launcher.py
1 parent c7f4297 commit b38bb3e

File tree

1 file changed

+1
-3
lines changed

1 file changed

+1
-3
lines changed

tools/accuracy_checker/accuracy_checker/launcher/pytorch_launcher.py

Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -16,7 +16,6 @@
1616

1717
from contextlib import contextmanager
1818
import sys
19-
import os
2019
import importlib
2120
import urllib
2221
import re
@@ -244,8 +243,7 @@ def predict(self, inputs, metadata=None, **kwargs):
244243
result_dict = { 'logits': output.logits.detach().cpu().numpy() }
245244
results.append(result_dict)
246245
continue
247-
else:
248-
outputs = list(output)
246+
outputs = list(output)
249247

250248
for meta_ in metadata:
251249
meta_['input_shape'] = {key: list(data.shape) for key, data in batch_input.items()}

0 commit comments

Comments
 (0)