diff --git a/ludwig/api.py b/ludwig/api.py
index c85fc67c3c5..d9ff9bd06d1 100644
--- a/ludwig/api.py
+++ b/ludwig/api.py
@@ -332,6 +332,7 @@ def __init__(
         # online training state
         self._online_trainer = None
 
+    # TODO: <Alex>ALEX</Alex>
     def train(
         self,
         dataset: Optional[Union[str, dict, pd.DataFrame]] = None,