Commit fc9c10c

committed

fix: handle zero total weight in cog function with ValueError

1 parent 67fe25d commit fc9c10cCopy full SHA for fc9c10c

File tree

-0

lines changed

-0

lines changed

Lines changed: 3 additions & 0 deletions

Original file line number	Diff line number	Diff line change
`@@ -20,7 +20,10 @@ def cog(target_weights: list[tuple[Set, float]]) -> float:`
`20`	`20`	`• xi is the corresponding value for the ith element in the output domain.`
`21`	`21`
`22`	`22`	`"""`
	`23`	`+`
`23`	`24`	`sum_weights = sum(weight for _, weight in target_weights)`
	`25`	`+ if sum_weights == 0:`
	`26`	`+ raise ValueError("Total weight is zero. Cannot compute center-of-gravity.")`
`24`	`27`	`sum_weighted_cogs = sum(then_set.center_of_gravity() * weight for then_set, weight in target_weights)`
`25`	`28`	`return sum_weighted_cogs / sum_weights`
`26`	`29`

Comments

(0)