note to self RE typecasting in the identity preprocessor -- not sure …

…this is desired behaviour always
GFNOrg · Nov 23, 2023 · f6edd53 · f6edd53
1 parent a4c1786
commit f6edd53
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/src/gfn/preprocessors.py b/src/gfn/preprocessors.py
@@ -31,7 +31,9 @@ class IdentityPreprocessor(Preprocessor):
     This is the default preprocessor used."""
 
     def preprocess(self, states: States) -> TT["batch_shape", "input_dim"]:
-        return states.tensor.float()
+        return (
+            states.tensor.float()
+        )  # TODO: should we typecast here? not a true identity...
 
 
 class EnumPreprocessor(Preprocessor):