Minor style and typing fix for gpustat.{cli,test_gpustat}

wookayin · Nov 24, 2023 · f4cd639 · f4cd639
1 parent f1464ea
commit f4cd639
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 4 deletions.
diff --git a/gpustat/cli.py b/gpustat/cli.py
@@ -1,3 +1,5 @@
+"""gpustat CLI."""
+
 import os
 import sys
 import time
@@ -104,15 +106,17 @@ def loop_gpustat(interval=1.0, **kwargs):
 
 
 def main(*argv):
+    """The main entrypoint to the gpustat CLI."""
     if not argv:
         argv = list(sys.argv)
 
     # attach SIGPIPE handler to properly handle broken pipe
     try:  # sigpipe not available under windows. just ignore in this case
         import signal
         signal.signal(signal.SIGPIPE, signal.SIG_DFL)
-    except Exception as e:
+    except Exception:  # pylint: disable=broad-exception-caught
         pass
+
     # arguments to gpustat
     import argparse
     try:
@@ -200,7 +204,7 @@ def nonnegative_int(value):
         args.show_fan_speed = True
         args.show_codec = 'enc,dec'
         args.show_power = 'draw,limit'
-    del args.show_all
+    del args.show_all  # type: ignore
 
     if args.interval is None:  # with default value
         args.interval = 1.0
@@ -213,7 +217,7 @@ def nonnegative_int(value):
 
         loop_gpustat(**vars(args))
     else:
-        del args.interval
+        del args.interval  # type: ignore
         print_gpustat(**vars(args))
 
 

diff --git a/gpustat/test_gpustat.py b/gpustat/test_gpustat.py
@@ -3,7 +3,7 @@
 """
 # flake8: ignore=E501
 # pyright: reportGeneralTypeIssues=false
-# pylint: disable=protected-access,no-member
+# pylint: disable=protected-access,no-member,redefined-outer-name
 
 import ctypes
 import os