Change directive, small update to example

Author: fourndo

SimPEG/directives/directives.py

@@ -1367,17 +1367,9 @@ def stack_channels(self, dpred: list):
         Regroup channel values along rows.
         """
         if isinstance(dpred, np.ndarray):
-            return dpred.reshape((1, -1))
+            return self.reshape(dpred)
 
-        tile_stack = []
-        channels = []
-        for pred in dpred:
-            channels += [pred]
-            if len(channels) == len(self.channels):
-                tile_stack += [self.reshape(np.vstack(channels))]
-                channels = []
-
-        return np.dstack(tile_stack)
+        return self.reshape(np.hstack(dpred))
 
     def save_components(self, iteration: int, values: list[np.ndarray] = None):
         """

tutorials/08-tdem/plot_inv_3_tem_3d.py

@@ -340,7 +340,7 @@ def reshape(values):
 data_object = data.Data(
     survey,
     dobs=dpred + noise,
-    noise_floor=floors.flatten(),
+    noise_floor=floors.flatten(order='F'),
     # noise_floor=np.kron(
     #     np.mean(np.abs(dpred).reshape), axis=1),
     #     np.ones(n_times),
@@ -453,7 +453,7 @@ def reshape(values):
 plt.plot(reshape(np.r_[inv_prob.dpred]).squeeze().T, "r")
 plt.plot(reshape(data_object.standard_deviation).squeeze().T, "k--")
 plt.plot(reshape(-data_object.standard_deviation).squeeze().T, "k--")
-axs.set_yscale("symlog", linthresh=1e-11)
+axs.set_yscale("symlog", linthresh=1e-13)
 axs.set_title("LP Predicted")
 axs.set_ylim([-1e-10, 0])
 
@@ -464,7 +464,7 @@ def reshape(values):
     plt.plot(reshape(simulation.dpred(inv_prob.l2model)).squeeze().T, "r")
     plt.plot(reshape(data_object.standard_deviation).squeeze().T, "k--")
     plt.plot(reshape(-data_object.standard_deviation).squeeze().T, "k--")
-    axs.set_yscale("symlog", linthresh=1e-11)
+    axs.set_yscale("symlog", linthresh=1e-14)
     axs.set_title("L2 Predicted")
     axs.set_ylim([-1e-10, 0])
     plt.show()