do not shadow built-in property (#841)

FAIR-Chem · Sep 13, 2024 · 83ab799 · 83ab799
1 parent 7d40b20
commit 83ab799
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/fairchem/core/trainers/ocp_trainer.py b/src/fairchem/core/trainers/ocp_trainer.py
@@ -258,8 +258,8 @@ def _forward(self, batch):
                     assert (
                         "property" in self.output_targets[target_key]
                     ), 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]}"
-                    property = self.output_targets[target_key]["property"]
-                    pred = out[target_key][property]
+                    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