fix decomposition in output logic

src/fairchem/core/trainers/ocp_trainer.py

@@ -260,10 +260,8 @@ def _forward(self, batch):
                     ), f"we need to know which property to match the target to, please specify the property field in the task config, current config: {self.output_targets[target_key]}"
                     prop = self.output_targets[target_key]["property"]
                     pred = out[target_key][prop]
-
-            ## TODO: deprecate the following logic?
-            ## Otherwise, assume target property is a derived output of the model. Construct the parent property
-            elif self.output_targets[target_key] == "decomposition":
+            # TODO clean up this logic to reconstruct a tensor from its predicted decomposition
+            elif "decomposition" in self.output_targets[target_key]:
                 _max_rank = 0
                 for subtarget_key in self.output_targets[target_key]["decomposition"]:
                     _max_rank = max(