fix standardize

Author: Kucharssim

bayesflow/adapters/adapter.py

@@ -108,8 +108,9 @@ def forward(
 
         log_det_jac = {}
         for transform in self.transforms:
+            transformed_data = transform(data, stage=stage, **kwargs)
             log_det_jac = transform.log_det_jac(data, log_det_jac, **kwargs)
-            data = transform(data, stage=stage, **kwargs)
+            data = transformed_data
 
         return data, log_det_jac
 

bayesflow/adapters/transforms/standardize.py

@@ -122,8 +122,6 @@ def inverse(self, data: np.ndarray, **kwargs) -> np.ndarray:
         return data * std + mean
 
     def log_det_jac(self, data, inverse: bool = False, **kwargs) -> np.ndarray:
-        if self.std is None:
-            return None
         std = np.broadcast_to(self.std, data.shape)
         ldj = np.log(np.abs(std))
         if inverse: