updated umato to deal with the bug according to the number of hubs

hyungkwonko · Apr 3, 2023 · ec9b6d4 · ec9b6d4
1 parent 0d2c0c6
commit ec9b6d4
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/src/umato/umato_.py b/src/umato/umato_.py
@@ -651,6 +651,8 @@ def __init__(
         self.ll = ll
 
     def _validate_parameters(self):
+        if self.hub_num >= self._raw_data.shape[0]:
+            raise ValueError("hub_num must be less than the number of data points")
         if self.set_op_mix_ratio < 0.0 or self.set_op_mix_ratio > 1.0:
             raise ValueError("set_op_mix_ratio must be between 0.0 and 1.0")
         if self.gamma < 0.0: