check for negative weights

throw valueerror
liedllab · Oct 5, 2021 · 4a7e25d · 4a7e25d
1 parent cb04da9
commit 4a7e25d
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/package/xentropy/internal/pre_post_processing.py b/package/xentropy/internal/pre_post_processing.py
@@ -29,6 +29,9 @@ def process_weights_argument(weights, verbose=False):
         if verbose:
             print("Weights have been given.")
         weights = np.array(weights)
+        if True in (weights < 0):
+            err_msg = "You gave negative values in you weights array.\n"
+            raise ValueError(err_msg)
 
     return weights, weight_switch