diff --git a/nvhtop.py b/nvhtop.py index bba0b94..d3ad358 100755 --- a/nvhtop.py +++ b/nvhtop.py @@ -1,33 +1,5 @@ #!/usr/bin/python3 -################################################################################# -# Copyright (c) 2019, NVIDIA Corporation. All rights reserved. # -# # -# Redistribution and use in source and binary forms, with or without # -# modification, are permitted provided that the following conditions are met: # -# # -# * Redistributions of source code must retain the above copyright notice, # -# this list of conditions and the following disclaimer. # -# * Redistributions in binary form must reproduce the above copyright # -# notice, this list of conditions and the following disclaimer in the # -# documentation and/or other materials provided with the distribution. # -# * Neither the name of the NVIDIA Corporation nor the names of its # -# contributors may be used to endorse or promote products derived from # -# this software without specific prior written permission. # -# # -# THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" # -# AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE # -# IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE # -# ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE # -# LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR # -# CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF # -# SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS # -# INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN # -# CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) # -# ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF # -# THE POSSIBILITY OF SUCH DAMAGE. # -################################################################################# - # To Run: # $ python3 nvhtop.py @@ -38,7 +10,6 @@ import time import psutil from cachetools import cached, TTLCache -from termcolor import colored import pynvml as nvml @@ -262,13 +233,7 @@ class Top(object): curses.endwin() for row in self.rows: if not isinstance(row, str): - row, attr = row - color = { - curses.color_pair(1): 'green', - curses.color_pair(2): 'yellow', - curses.color_pair(3): 'red', - }.get(attr) - row = colored(row, color) + row = row[0] print(row) def init_curses(self): diff --git a/requirements.txt b/requirements.txt index 85a6f56..17321e8 100644 --- a/requirements.txt +++ b/requirements.txt @@ -1,4 +1,4 @@ nvidia-ml-py psutil cachetools -termcolor +windows-curses; sys_platform == 'windows'