Merge pull request #11 from cspencerjones/single-chunk

Reshape data better in the numpy layer
cspencerjones · May 12, 2020 · 2ed10e8 · 2ed10e8
2 parents 1b2d36b + d4ddd70
commit 2ed10e8
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/xlayers/core.py b/xlayers/core.py
@@ -166,7 +166,7 @@ def layers_numpy(v_in, theta_in, thetalayers, mapfact, mapindex, cellindex, drf_
     new_shape = list(original_shape)
     new_shape[-1] = thetalayers.size
     new_shape = tuple(new_shape)
-    v_lay = np.squeeze(np.array(_reshape_outputs(VH2, shape=new_shape)))
+    v_lay = np.squeeze(np.array(_reshape_outputs(VH2, shape=new_shape)),axis=0)
 
     return v_lay