From 3d52281b51c77038cf6e0d7cd37df7831cb7f806 Mon Sep 17 00:00:00 2001
From: AdrienTaylor <10559960+AdrienTaylor@users.noreply.github.com>
Date: Fri, 7 Nov 2025 07:44:19 +0100
Subject: [PATCH] fix int overflow

---
 PEPit/wrappers/mosek_wrapper.py | 4 ++--
 docs/source/whatsnew/0.4.1.rst  | 4 ++++
 2 files changed, 6 insertions(+), 2 deletions(-)
 create mode 100644 docs/source/whatsnew/0.4.1.rst

diff --git a/PEPit/wrappers/mosek_wrapper.py b/PEPit/wrappers/mosek_wrapper.py
index be89fa2c..1cc2427d 100644
--- a/PEPit/wrappers/mosek_wrapper.py
+++ b/PEPit/wrappers/mosek_wrapper.py
@@ -146,7 +146,7 @@ def send_constraint_to_solver(self, constraint, track=True):
 
         sym_A = self.task.appendsparsesymmat(Point.counter, A_i, A_j, A_val)
         self.task.putbaraij(nb_cons, 0, [sym_A], [1.0])
-        self.task.putaijlist(nb_cons + np.zeros(a_i.shape, dtype=np.int8), a_i, a_val)
+        self.task.putaijlist(np.full(a_i.shape, nb_cons), a_i, a_val)
 
         if track:
             self._constraint_index_in_mosek.append(nb_cons)
@@ -201,7 +201,7 @@ def send_lmi_constraint_to_solver(self, psd_counter, psd_matrix):
                 # fill the mosek (equality) constraint 
                 self.task.putbaraij(nb_cons, 0, [sym_A1], [1.0])
                 self.task.putbaraij(nb_cons, psd_matrix.counter + 1, [sym_A2], [1.0])
-                self.task.putaijlist(nb_cons + np.zeros(a_i.shape, dtype=np.int8), a_i, a_val)
+                self.task.putaijlist(np.full(a_i.shape, nb_cons), a_i, a_val)
                 self.task.putconbound(nb_cons, mosek.boundkey.fx, -alpha_val, -alpha_val)
 
         # Print a message if verbose mode activated
diff --git a/docs/source/whatsnew/0.4.1.rst b/docs/source/whatsnew/0.4.1.rst
new file mode 100644
index 00000000..7f677b42
--- /dev/null
+++ b/docs/source/whatsnew/0.4.1.rst
@@ -0,0 +1,4 @@
+What's new in PEPit 0.4.1
+=========================
+
+- Fix: An integer overflow (due to previously specified dtype=int8) in :class:`MosekWrapper` has been fixed.