From 8c6a02906d2cf292bc4cf08822af39893b0024d0 Mon Sep 17 00:00:00 2001
From: Kyle Gorman <kylebgorman@gmail.com>
Date: Sat, 30 Nov 2024 13:20:32 -0500
Subject: [PATCH] Expands list of separate features models

---
 yoyodyne/predict.py | 8 ++++++--
 1 file changed, 6 insertions(+), 2 deletions(-)

diff --git a/yoyodyne/predict.py b/yoyodyne/predict.py
index 15a8b000..e4154864 100644
--- a/yoyodyne/predict.py
+++ b/yoyodyne/predict.py
@@ -36,9 +36,13 @@ def get_datamodule_from_argparse_args(
         data.DataModule.
     """
     separate_features = args.features_col != 0 and args.arch in [
-        "pointer_generator_rnn",
+        "hard_attention_gru",
+        "hard_attention_lstm",
+        "pointer_generator_gru",
+        "pointer_generator_lstm",
         "pointer_generator_transformer",
-        "transducer",
+        "transducer_grm",
+        "transducer_lstm",
     ]
     index = data.Index.read(args.model_dir)
     return data.DataModule(