Skip to content

Commit 85b1cb0

Browse files
committed
fmt
1 parent 044e845 commit 85b1cb0

File tree

5 files changed

+35
-26
lines changed

5 files changed

+35
-26
lines changed

src/ol_openedx_course_translations/ol_openedx_course_translations/filters.py

Lines changed: 0 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -7,8 +7,6 @@
77

88
from ol_openedx_course_translations.utils.constants import (
99
ENGLISH_LANGUAGE_CODE,
10-
ES_419_LANGUAGE_CODE,
11-
ES_LANGUAGE_CODE,
1210
)
1311

1412
VIDEO_BLOCK_TYPE = "video"

src/ol_openedx_course_translations/ol_openedx_course_translations/management/commands/translate_course.py

Lines changed: 4 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -18,8 +18,7 @@
1818
translate_policy_json_task,
1919
)
2020
from ol_openedx_course_translations.utils.constants import (
21-
ES_419_LANGUAGE_CODE,
22-
ES_LANGUAGE_CODE,
21+
ENGLISH_LANGUAGE_CODE,
2322
PROVIDER_DEEPL,
2423
)
2524
from ol_openedx_course_translations.utils.course_translations import (
@@ -32,8 +31,6 @@
3231
validate_course_inputs,
3332
)
3433

35-
from ol_openedx_course_translations.utils.constants import ENGLISH_LANGUAGE_CODE
36-
3734
logger = logging.getLogger(__name__)
3835

3936
# Task configuration
@@ -231,7 +228,7 @@ def _parse_and_validate_provider_spec(
231228

232229
return provider_name, default_model
233230

234-
def handle(self, **options) -> None: # noqa: PLR0915
231+
def handle(self, **options) -> None: # noqa: PLR0915, PLR0912, C901
235232
"""Handle the translate_course command."""
236233
try:
237234
start_time = time.perf_counter()
@@ -252,14 +249,14 @@ def handle(self, **options) -> None: # noqa: PLR0915
252249
f"Source language '{source_language}' is not supported. "
253250
f"Only `en` is supported as source language at this time."
254251
)
255-
raise CommandError(error_msg)
252+
raise CommandError(error_msg) # noqa: TRY301
256253

257254
if target_language not in settings.COURSE_TRANSLATIONS_SUPPORTED_LANGUAGES:
258255
error_msg = (
259256
f"Target language '{target_language}' is not supported. "
260257
f"Supported languages: {', '.join(settings.COURSE_TRANSLATIONS_SUPPORTED_LANGUAGES.keys())}" # noqa: E501
261258
)
262-
raise CommandError(error_msg)
259+
raise CommandError(error_msg) # noqa: TRY301
263260

264261
content_provider_spec = options["content_translation_provider"]
265262
srt_provider_spec = options["srt_translation_provider"]

src/ol_openedx_course_translations/ol_openedx_course_translations/providers/llm_providers.py

Lines changed: 28 additions & 14 deletions
Original file line numberDiff line numberDiff line change
@@ -159,8 +159,10 @@ def _get_text_system_prompt(
159159
load_glossary,
160160
)
161161

162-
target_language_display_name = settings.COURSE_TRANSLATIONS_SUPPORTED_LANGUAGES.get(
163-
target_language, target_language
162+
target_language_display_name = (
163+
settings.COURSE_TRANSLATIONS_SUPPORTED_LANGUAGES.get(
164+
target_language, target_language
165+
)
164166
)
165167

166168
system_prompt = (
@@ -408,8 +410,10 @@ def make_payload(chunk_idxs: list[int]) -> str:
408410
return "\n".join(parts)
409411

410412
# Prompt specifically for batches of plain strings
411-
target_language_display_name = settings.COURSE_TRANSLATIONS_SUPPORTED_LANGUAGES.get(
412-
target_language, target_language
413+
target_language_display_name = (
414+
settings.COURSE_TRANSLATIONS_SUPPORTED_LANGUAGES.get(
415+
target_language, target_language
416+
)
413417
)
414418
system_prompt = (
415419
f"You are a professional translator. "
@@ -789,11 +793,15 @@ def _get_translation_validation_system_prompt(
789793
source_language: str,
790794
target_language: str,
791795
) -> str:
792-
target_language_display_name = settings.COURSE_TRANSLATIONS_SUPPORTED_LANGUAGES.get(
793-
target_language, target_language
796+
target_language_display_name = (
797+
settings.COURSE_TRANSLATIONS_SUPPORTED_LANGUAGES.get(
798+
target_language, target_language
799+
)
794800
)
795-
source_language_display_name = settings.COURSE_TRANSLATIONS_SUPPORTED_LANGUAGES.get(
796-
source_language, source_language
801+
source_language_display_name = (
802+
settings.COURSE_TRANSLATIONS_SUPPORTED_LANGUAGES.get(
803+
source_language, source_language
804+
)
797805
)
798806

799807
return (
@@ -924,8 +932,10 @@ def _get_subtitle_system_prompt(
924932
Returns:
925933
System prompt string for subtitle translation
926934
"""
927-
target_language_display_name = settings.COURSE_TRANSLATIONS_SUPPORTED_LANGUAGES.get(
928-
target_language, target_language
935+
target_language_display_name = (
936+
settings.COURSE_TRANSLATIONS_SUPPORTED_LANGUAGES.get(
937+
target_language, target_language
938+
)
929939
)
930940

931941
# V1
@@ -1050,8 +1060,10 @@ def _get_subtitle_system_prompt(
10501060
Returns:
10511061
System prompt string for subtitle translation
10521062
"""
1053-
target_language_display_name = settings.COURSE_TRANSLATIONS_SUPPORTED_LANGUAGES.get(
1054-
target_language, target_language
1063+
target_language_display_name = (
1064+
settings.COURSE_TRANSLATIONS_SUPPORTED_LANGUAGES.get(
1065+
target_language, target_language
1066+
)
10551067
)
10561068

10571069
system_prompt = (
@@ -1140,8 +1152,10 @@ def _get_subtitle_system_prompt(
11401152
Returns:
11411153
System prompt string for subtitle translation
11421154
"""
1143-
target_language_display_name = settings.COURSE_TRANSLATIONS_SUPPORTED_LANGUAGES.get(
1144-
target_language, target_language
1155+
target_language_display_name = (
1156+
settings.COURSE_TRANSLATIONS_SUPPORTED_LANGUAGES.get(
1157+
target_language, target_language
1158+
)
11451159
)
11461160

11471161
system_prompt = (

src/ol_openedx_course_translations/ol_openedx_course_translations/tasks.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -144,7 +144,9 @@ def translate_file_task( # noqa: PLR0913, PLR0912, C901
144144
"reason": "Not source language SRT",
145145
}
146146

147-
output_filename = get_srt_output_filename(file_path.name, target_language.lower().replace("_", "-"))
147+
output_filename = get_srt_output_filename(
148+
file_path.name, target_language.lower().replace("_", "-")
149+
)
148150
output_file_path = file_path.parent / output_filename
149151

150152
provider.translate_document(

src/ol_openedx_course_translations/ol_openedx_course_translations/utils/course_translations.py

Lines changed: 0 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -32,8 +32,6 @@
3232
OpenAIProvider,
3333
)
3434
from ol_openedx_course_translations.utils.constants import (
35-
ES_419_LANGUAGE_CODE,
36-
ES_LANGUAGE_CODE,
3735
NEVER_TRANSLATE_ATTRS,
3836
PROVIDER_DEEPL,
3937
PROVIDER_GEMINI,

0 commit comments

Comments
 (0)