Fixes bug in K_repr_ax

cvxgrp · Apr 30, 2024 · 684d0d5 · 684d0d5
1 parent 61a66d3
commit 684d0d5
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/dsp/cone_transforms.py b/dsp/cone_transforms.py
@@ -204,10 +204,10 @@ def K_repr_ax(a: cp.Expression) -> KRepresentation:
     assert a.is_convex()
     assert a.size == 1
 
-    f = cp.Constant(0)
+    f = cp.Variable(name="f_ax")
     t = cp.Variable(name="t_ax")
 
-    constraints = [t >= a]
+    constraints = [t >= a, f == 0]
 
     return KRepresentation(f=f, t=t, constraints=constraints, concave_expr=lambda x: a.value)