Quick fixes for failing tests for distribution regression (#40)

neuralmagic · Aug 28, 2024 · 9cb19d9 · 9cb19d9
1 parent 0719029
commit 9cb19d9
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/guidellm/core/distribution.py b/src/guidellm/core/distribution.py
@@ -120,7 +120,7 @@ def min(self) -> float:
             logger.warning("No data points available to calculate minimum.")
             return 0.0
 
-        min_value: float = np.min(self.data)
+        min_value: float = np.min(self.data).item()  # type: ignore  # noqa: PGH003
         logger.debug(f"Calculated min: {min_value}")
         return min_value
 
@@ -134,7 +134,7 @@ def max(self) -> float:
             logger.warning("No data points available to calculate maximum.")
             return 0.0
 
-        max_value: float = np.max(self.data)
+        max_value: float = np.max(self.data).item()  # type: ignore  # noqa: PGH003
         logger.debug(f"Calculated max: {max_value}")
         return max_value