MAINT: Avoid future issue in NumPy

bashtage · Jan 4, 2024 · 123893f · 123893f
1 parent 0502dd7
commit 123893f
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/linearmodels/panel/model.py b/linearmodels/panel/model.py
@@ -1878,7 +1878,9 @@ def fit(
                     if self._constant:
                         assert isinstance(self._constant_index, int)
                         self._constant_index = int(
-                            np.argwhere(np.array(retain) == self._constant_index)
+                            np.squeeze(
+                                np.argwhere(np.array(retain) == self._constant_index)
+                            )
                         )
 
                     # Adjust exog