Skip to content

Commit f26413f

Browse files
committed
Fix issue with NaN values for GPUs
1 parent 4b3ea81 commit f26413f

File tree

1 file changed

+4
-2
lines changed

1 file changed

+4
-2
lines changed

library/stats.py

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,3 +1,5 @@
1+
import math
2+
13
import GPUtil
24
import psutil
35

@@ -185,7 +187,7 @@ def display_gpu_stats(load, memory_percentage, memory_used, temperature):
185187
y=THEME_DATA['STATS']['GPU']['PERCENTAGE']['GRAPH'].get("Y", 0),
186188
width=THEME_DATA['STATS']['GPU']['PERCENTAGE']['GRAPH'].get("WIDTH", 0),
187189
height=THEME_DATA['STATS']['GPU']['PERCENTAGE']['GRAPH'].get("HEIGHT", 0),
188-
value=int(load),
190+
value=int(load) if not math.isnan(load) else 0,
189191
min_value=THEME_DATA['STATS']['GPU']['PERCENTAGE']['GRAPH'].get("MIN_VALUE", 0),
190192
max_value=THEME_DATA['STATS']['GPU']['PERCENTAGE']['GRAPH'].get("MAX_VALUE", 100),
191193
bar_color=THEME_DATA['STATS']['GPU']['PERCENTAGE']['GRAPH'].get("BAR_COLOR", (0, 0, 0)),
@@ -198,7 +200,7 @@ def display_gpu_stats(load, memory_percentage, memory_used, temperature):
198200

199201
if THEME_DATA['STATS']['GPU']['PERCENTAGE']['TEXT'].get("SHOW", False):
200202
display.lcd.DisplayText(
201-
text=f"{int(load):>3}%",
203+
text=f"{int(load):>3}%" if not math.isnan(load) else f"{int(0):>3}%",
202204
x=THEME_DATA['STATS']['GPU']['PERCENTAGE']['TEXT'].get("X", 0),
203205
y=THEME_DATA['STATS']['GPU']['PERCENTAGE']['TEXT'].get("Y", 0),
204206
font=THEME_DATA['STATS']['GPU']['PERCENTAGE']['TEXT'].get("FONT", "roboto-mono/RobotoMono-Regular.ttf"),

0 commit comments

Comments
 (0)