Fix wrong Y axis scale in Aim UI when metric has infinity values (#494)

G-Research · Oct 19, 2023 · c616d3d · c616d3d
1 parent b24c6a0
commit c616d3d
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/pkg/api/aim/runs.go b/pkg/api/aim/runs.go
@@ -1032,6 +1032,12 @@ func DeleteBatch(c *fiber.Ctx) error {
 func toNumpy(values []float64) fiber.Map {
 	buf := bytes.NewBuffer(make([]byte, 0, len(values)*8))
 	for _, v := range values {
+		switch v {
+		case math.MaxFloat64:
+			v = math.Inf(1)
+		case -math.MaxFloat64:
+			v = math.Inf(-1)
+		}
 		binary.Write(buf, binary.LittleEndian, v)
 	}
 	return fiber.Map{