added TODO

GFNOrg · Jan 15, 2025 · fe9713c · fe9713c
1 parent 4ee6987
commit fe9713c
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/gfn/preprocessors.py b/src/gfn/preprocessors.py
@@ -77,7 +77,7 @@ def preprocess(self, states) -> torch.Tensor:
 
 class GraphPreprocessor(Preprocessor):
     def __init__(self) -> None:
-        super().__init__(-1)
+        super().__init__(-1)  # TODO: review output_dim API
 
     def preprocess(self, states: GraphStates) -> TensorDict:
         return states.tensor