fix squeeze in distances for WellModel

pastas · Feb 5, 2025 · 2cdc095 · 2cdc095
1 parent 7b804e4
commit 2cdc095
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/pastastore/store.py b/pastastore/store.py
@@ -1173,7 +1173,7 @@ def get_stressmodel(
             # compute distances and add to kwargs
             distances = (
                 self.get_distances(oseries=oseries, stresses=names)
-                .T.squeeze(axis=0)
+                .T.squeeze(axis=1)
                 .values
             )
             kwargs["distances"] = distances