Features fix

chemprop · Sep 12, 2020 · 3472968 · 3472968
1 parent d16b4fc
commit 3472968
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/chemprop/data/data.py b/chemprop/data/data.py
@@ -98,7 +98,7 @@ def __init__(self,
             self.atom_features = np.where(np.isnan(self.atom_features), replace_token, self.atom_features)
 
         # Save a copy of the raw features and targets to enable different scaling later on
-        self.raw_features, self.raw_targets = features, targets
+        self.raw_features, self.raw_targets = self.features, self.targets
 
     @property
     def mol(self) -> Chem.Mol: