fix: backtranslation with api and README

iarahealth · Nov 27, 2023 · e6bb563 · e6bb563
1 parent 211e640
commit e6bb563
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 5 deletions.
diff --git a/README.md b/README.md
@@ -8,7 +8,7 @@ Can be used standalone (instructions below) or as part of other programs.
 help: ./run_txt_aug.py -h
 
 example usage:
-./run_txt_aug.py corpus_1br_10pt_15sept.tok --aug translate random --action delete --maxs 10 --lang en --translate_mode local --append --output out.txt
+./run_txt_aug.py corpus_1br_10pt_15sept.tok --aug backtranslate random_del --maxs 5 --lang en --translate_mode google --apend --output out.txt
 ```
 
 ## Offline text generation

diff --git a/run_txt_aug.py b/run_txt_aug.py
@@ -11,7 +11,7 @@
 
 """
 example usage:
-./txt_aug.py corpus.tok --aug translate random --action delete --maxs 10 --lang en --append
+./txt_aug.py corpus.tok --aug translate random_del --maxs 10 --lang en --append
 """
 
 

diff --git a/text/aug.py b/text/aug.py
@@ -223,14 +223,13 @@ def backtranslate_sentences_api(
  skips_count += 1
  continue
  try:
- backtranslator = backtranslator.translate(translation)
+ backtranslation = backtranslator.translate(translation)
+ backtranslations.append(backtranslation)
  except Exception as e:
  print(f"[!] Skipping backtranslating sentence '{translation}': {e}")
  skips_count += 1
  continue
  skips_count = 0
- backtranslation = backtranslator.translate(translation)
- backtranslations.append(backtranslation)
  print(f"{sentence} -> {translation} -> {backtranslation}")
 
  return backtranslations