Updates based on reviewer comments (#1)

* Fix variable definition in wrong place

If the data variable is defined where it was, the code failed if
transform_grayscale=false. Since we were running with true value,
this did not impact us before.

* Improve plotting of rain rate differences

* Improved plotting based on reviewer comments

- Add option to plot background map & rain rate difference in case figs
- Add script to plot dataset value distributions

* Add stylefile for distributions script

datasets/LagrangianFMIComposite.py

@@ -280,10 +280,10 @@ def invScaler(self, data: torch.Tensor):
             return data
 
     def postprocessing(self, data_in: np.ndarray):
+        data = torch.Tensor(data_in)
         if self.transform_to_grayscale:
             # data of shape (window_size, im.shape[0], im.shape[1])
             # dbZ to mm/h
-            data = torch.Tensor(data_in)
             data = 10 ** (data * 0.1)
             data = (data / 223) ** (1 / 1.53)  # fixed