Skip to content

Commit 3169a9b

Browse files
author
igor
committed
Fix for linter
1 parent 9fed358 commit 3169a9b

File tree

1 file changed

+4
-2
lines changed

1 file changed

+4
-2
lines changed

batchflow/models/torch/base.py

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1833,7 +1833,8 @@ def load(self, file, make_infrastructure=False, mode='eval', pickle_module=dill,
18331833
self.set_model_mode(mode)
18341834

18351835
return
1836-
elif file.endswith(".onnx"):
1836+
1837+
if file.endswith(".onnx"):
18371838
try:
18381839
from onnx2torch import convert
18391840
except ImportError as e:
@@ -1850,7 +1851,8 @@ def load(self, file, make_infrastructure=False, mode='eval', pickle_module=dill,
18501851
self.set_model_mode(mode)
18511852

18521853
return
1853-
elif file.endswith(".openvino"):
1854+
1855+
if file.endswith(".openvino"):
18541856
model = OVModel(model_path=file, **model_load_kwargs)
18551857
self.model = model
18561858

0 commit comments

Comments
 (0)