Skip to content

Commit 4a3732e

Browse files
committed
Fix mypy errors
1 parent 1db4060 commit 4a3732e

File tree

4 files changed

+3
-24
lines changed

4 files changed

+3
-24
lines changed

argostranslate/apply_bpe.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -415,7 +415,7 @@ def isolate_glossary(word, glossary):
415415
if __name__ == "__main__":
416416

417417
currentdir = os.path.dirname(
418-
os.path.abspath(inspect.getfile(inspect.currentframe()))
418+
os.path.abspath(inspect.getfile(inspect.currentframe())) # type: ignore
419419
)
420420
newdir = os.path.join(currentdir, "subword_nmt")
421421
if os.path.isdir(newdir):

argostranslate/sbd.py

+2-2
Original file line numberDiff line numberDiff line change
@@ -85,7 +85,7 @@ def parse_fewshot_response(response_text: str) -> str | None:
8585
return response[-1]
8686

8787

88-
def process_seq2seq_sbd(input_text: str, sbd_translated_guess: str) -> int:
88+
def process_seq2seq_sbd(input_text: str, sbd_translated_guess: str) -> int | None:
8989
sbd_translated_guess_index = sbd_translated_guess.find(SENTENCE_BOUNDARY_TOKEN)
9090
if sbd_translated_guess_index != -1:
9191
sbd_translated_guess = sbd_translated_guess[:sbd_translated_guess_index]
@@ -107,7 +107,7 @@ def process_seq2seq_sbd(input_text: str, sbd_translated_guess: str) -> int:
107107

108108
def detect_sentence(
109109
input_text: str, sbd_translation, sentence_guess_length: int = 150
110-
) -> int:
110+
) -> int | None:
111111
"""Given input text, return the index after the end of the first sentence.
112112
113113
Args:

argostranslate/translate.py

-16
Original file line numberDiff line numberDiff line change
@@ -466,22 +466,6 @@ def get_installed_languages() -> list[Language]:
466466
)
467467
from_lang.translations_from.append(remote_translation)
468468
to_lang.translations_to.append(remote_translation)
469-
elif (
470-
argostranslate.settings.model_provider
471-
== argostranslate.settings.ModelProvider.OPENAI
472-
):
473-
language_model = argostranslate.apis.OpenAIAPI(
474-
argostranslate.settings.openai_api_key
475-
)
476-
# TODO
477-
languages = [Language("en", "English"), Language("es", "Spanish")]
478-
for from_lang in languages:
479-
for to_lang in languages:
480-
few_shot_translation = FewShotTranslation(
481-
from_lang, to_lang, language_model
482-
)
483-
from_lang.translations_from.append(few_shot_translation)
484-
to_lang.translations_to.append(few_shot_translation)
485469

486470
info("languages", languages)
487471

argostranslate/utils.py

-5
Original file line numberDiff line numberDiff line change
@@ -16,11 +16,6 @@
1616
# https://docs.python.org/3/library/logging.html#levels
1717

1818

19-
def debug(*argv):
20-
"""Debug level log"""
21-
logger.debug(str(argv))
22-
23-
2419
def get_json_string(argv):
2520
if len(argv) == 1:
2621
return json.dumps(str(argv[0]))

0 commit comments

Comments
 (0)