Fix pylint warnings

wxz1996 · Aug 25, 2020 · cb82ebf · cb82ebf
1 parent a845656
commit cb82ebf
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/compressai/utils/bench/codecs.py b/compressai/utils/bench/codecs.py
@@ -462,17 +462,17 @@ def get_vtm_encoder_path(build_dir):
     try:
         elfnames = {'Darwin': 'EncoderApp', 'Linux': 'EncoderAppStatic'}
         return os.path.join(build_dir, elfnames[system])
-    except KeyError:
-        raise RuntimeError(f'Unsupported platform "{system}"')
+    except KeyError as err:
+        raise RuntimeError(f'Unsupported platform "{system}"') from err
 
 
 def get_vtm_decoder_path(build_dir):
     system = platform.system()
     try:
         elfnames = {'Darwin': 'DecoderApp', 'Linux': 'DecoderAppStatic'}
         return os.path.join(build_dir, elfnames[system])
-    except KeyError:
-        raise RuntimeError(f'Unsupported platform "{system}"')
+    except KeyError as err:
+        raise RuntimeError(f'Unsupported platform "{system}"') from err
 
 
 class VTM(Codec):