fix: presample only when there are weights to pick

viktordanov · Oct 29, 2023 · 1eaf508 · 1eaf508
1 parent 1eaf8ea
commit 1eaf508
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/production_rule.go b/production_rule.go
@@ -118,6 +118,9 @@ func (bp *ByteProductionRule) RandomizeWeights(delta float64) {
 }
 
 func (bp *ByteProductionRule) PreSample() {
+	if bp.Weights == nil || len(bp.Weights) == 0 {
+		return
+	}
 	if bp.PreSampledWeights == nil {
 		bp.PreSampledWeights = make([]uint8, 256, 256)
 	}