fix: added model_task argument for correct config in retro task + upd…

…ate deps utils

setup.cfg

@@ -30,7 +30,7 @@ install_requires =
     rxn-chem-utils>=1.1.4
     rxn-reaction-preprocessing>=2.0.2
     rxn-utils>=1.1.9
-    rxn-onmt-utils @ git+https://github.com/rxn4chemistry/rxn-onmt-utils.git@f1a0b970411aac308a3cba36c942297933a4dd91 #rxn-onmt-utils without rxn-opennmt-py depedency
+    rxn-onmt-utils @ git+https://github.com/rxn4chemistry/rxn-onmt-utils.git@6725e6dad0c3fc563aa3eba498289c207fe9b8e2 #rxn-onmt-utils without rxn-opennmt-py depedency
     OpenNMT-py>=3.5.1 # official onmt
 
 [options.packages.find]

src/rxn/onmt_models/scripts/rxn_onmt_continue_training.py

@@ -57,6 +57,7 @@
     default=100000,
     help="Number of steps, including steps from the initial training run.",
 )
+@click.option("--model_task", type=str, required=True)
 def main(
     batch_size: int,
     data_weights: Tuple[int, ...],
@@ -66,6 +67,7 @@ def main(
     preprocess_dir: str,
     train_from: Optional[str],
     train_num_steps: int,
+    model_task: str,
 ) -> None:
     """Continue training for an OpenNMT model.
 
@@ -111,6 +113,7 @@ def main(
         train_steps=train_num_steps,
         no_gpu=no_gpu,
         data_weights=data_weights,
+        model_task=model_task,
     )
 
     # Write config file

src/rxn/onmt_models/scripts/rxn_onmt_finetune.py

@@ -54,6 +54,7 @@
 @click.option("--warmup_steps", default=defaults.WARMUP_STEPS)
 @click.option("--report_every", default=1000)
 @click.option("--save_checkpoint_steps", default=5000)
+@click.option("--model_task", type=str, required=True)
 def main(
     batch_size: int,
     data_weights: Tuple[int, ...],
@@ -69,6 +70,7 @@ def main(
     warmup_steps: int,
     report_every: int,
     save_checkpoint_steps: int,
+    model_task: str,
 ) -> None:
     """Finetune an OpenNMT model."""
 
@@ -122,6 +124,7 @@ def main(
         data_weights=data_weights,
         report_every=report_every,
         save_checkpoint_steps=save_checkpoint_steps,
+        model_task=model_task,
     )
 
     # Write config file

src/rxn/onmt_models/scripts/rxn_onmt_train.py

@@ -76,6 +76,7 @@ def check_rnn_vs_hidden_size(hidden_size: int, rnn_size: int) -> int:
 @click.option("--transformer_ff", default=defaults.TRANSFORMER_FF)
 @click.option("--warmup_steps", default=defaults.WARMUP_STEPS)
 @click.option("--word_vec_size", default=defaults.WORD_VEC_SIZE)
+@click.option("--model_task", type=str, required=True)
 def main(
     batch_size: int,
     data_weights: Tuple[int, ...],
@@ -94,6 +95,7 @@ def main(
     transformer_ff: int,
     warmup_steps: int,
     word_vec_size: int,
+    model_task: str,
 ) -> None:
     """Train an OpenNMT model.
 
@@ -143,6 +145,7 @@ def main(
         word_vec_size=word_vec_size,
         no_gpu=no_gpu,
         data_weights=data_weights,
+        model_task=model_task,
     )
 
     # Write config file