modified: auton_survival/estimators.py

autonlab · Mar 29, 2022 · e347556 · e347556
1 parent 6e542b3
commit e347556
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/auton_survival/estimators.py b/auton_survival/estimators.py
@@ -669,8 +669,8 @@ def fit(self, features, outcomes,
 
     if weights is not None:
       assert len(weights) == features.shape[0], "Size of passed weights must match size of training data."
+      assert (weights>0.).any(), "All weights must be positive."
       # assert ((weights>0.0)&(weights<=1.0)).all(), "Weights must be in the range (0,1]."
-
       # weights[weights>(1-weights_clip)] = 1-weights_clip
       # weights[weights<(weights_clip)] = weights_clip