diff --git a/heimdallr/utilities/nvidia/smi_parsing.py b/heimdallr/utilities/nvidia/smi_parsing.py index 1d7a922..b0329c4 100644 --- a/heimdallr/utilities/nvidia/smi_parsing.py +++ b/heimdallr/utilities/nvidia/smi_parsing.py @@ -2827,9 +2827,9 @@ def DeviceQuery(self, filter=None): or NVSMI_CLOCK_THROTTLE_REASONS_SW_THERMAL_SLOWDOWN in filter or NVSMI_CLOCK_THROTTLE_REASONS_SYNC_BOOST in filter ): - gpuResults[ - "clocks_throttle" - ] = NvidiaSMI.__GetClocksThrottleReasons(handle) + gpuResults["clocks_throttle"] = ( + NvidiaSMI.__GetClocksThrottleReasons(handle) + ) fbMemoryUsage = {} includeMemoryUsage = False @@ -3268,9 +3268,9 @@ def DeviceQuery(self, filter=None): except NVMLError as err: supportedGraphicsClocks = NvidiaSMI.__handleError(err) - supportMemClock[ - "supported_graphics_clock" - ] = supportedGraphicsClocks + supportMemClock["supported_graphics_clock"] = ( + supportedGraphicsClocks + ) supportedClocks.append(supportMemClock) # jj+=1 diff --git a/requirements.txt b/requirements.txt index 3108714..652bd28 100644 --- a/requirements.txt +++ b/requirements.txt @@ -11,4 +11,6 @@ schedule sorcery setuptools>=65.5.1 # not directly required, pinned by Snyk to avoid a vulnerability dash -plotly \ No newline at end of file +plotly +requests>=2.32.0 # not directly required, pinned by Snyk to avoid a vulnerability +werkzeug>=3.0.3 # not directly required, pinned by Snyk to avoid a vulnerability \ No newline at end of file