fix: c when custom weights are given

yoyolicoris · May 27, 2024 · af67a00 · af67a00
1 parent 7d7267c
commit af67a00
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/kamui/core.py b/kamui/core.py
@@ -101,7 +101,7 @@ def calculate_k(
         else:
             c = np.ones((M * 2,), dtype=np.int64)
     else:
-        c = np.tile(c, 2)
+        c = np.tile(weights, 2)
 
     res = linprog(c, A_eq=A_eq, b_eq=b_eq, integrality=1)
     k = res.x[:M] - res.x[M:]