Skip to content

Commit 9cb19d9

Browse files
authored
Quick fixes for failing tests for distribution regression (#40)
1 parent 0719029 commit 9cb19d9

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

src/guidellm/core/distribution.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -120,7 +120,7 @@ def min(self) -> float:
120120
logger.warning("No data points available to calculate minimum.")
121121
return 0.0
122122

123-
min_value: float = np.min(self.data)
123+
min_value: float = np.min(self.data).item() # type: ignore # noqa: PGH003
124124
logger.debug(f"Calculated min: {min_value}")
125125
return min_value
126126

@@ -134,7 +134,7 @@ def max(self) -> float:
134134
logger.warning("No data points available to calculate maximum.")
135135
return 0.0
136136

137-
max_value: float = np.max(self.data)
137+
max_value: float = np.max(self.data).item() # type: ignore # noqa: PGH003
138138
logger.debug(f"Calculated max: {max_value}")
139139
return max_value
140140

0 commit comments

Comments
 (0)