fix(utils.py::assign_layers): bad logic in return branch

DOI-USGS · Aug 28, 2024 · 45e45e6 · 45e45e6
1 parent fc2abd6
commit 45e45e6
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/sfrmaker/utils.py b/sfrmaker/utils.py
@@ -113,7 +113,7 @@ def assign_layers(reach_data, botm_array,
             botm_array = new_botm_array
         reach_data['k'] = layers
     elif return_3d or new_botm_array is None:
-        return layers, new_botm_array[-1]
+        return layers, new_botm_array
     else:
         warnings.warn('Previously, assign_layers has returned a 2D (model bottom) array. '
                       'Now, if idomain is specified, a 3D array of model cell bottoms is returned. '