improve dataframe iteration and register command

hyperskill · Mar 4, 2024 · e9e5071 · e9e5071
1 parent c99e3cb
commit e9e5071
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/jba/src/processing/tasktracker_platform_step_mapping.py b/jba/src/processing/tasktracker_platform_step_mapping.py
@@ -63,8 +63,7 @@ def find_mapping_solutions(user_df: pd.DataFrame, platform_df: pd.DataFrame, les
             break
 
     if last_status == EduTaskStatus.WRONG.value:
-        for index, _ in enumerate(range(user_ind, len(user_df))):
-            user_df.at[user_index[index + user_ind], TASKTRACKER_TASK_NAME] = platform_task_name
+        user_df.at[user_index[user_ind:len(user_df)], TASKTRACKER_TASK_NAME] = platform_task_name
 
     return user_df
 

diff --git a/pyproject.toml b/pyproject.toml
@@ -86,6 +86,7 @@ collect_course_structure = 'jba.src.processing.collect_course_structure:main'
 tasktracker_content_collector = 'jba.src.processing.tasktracker_content_collector:main'
 tasktracker_task_filter = 'jba.src.processing.tasktracker_task_filter:main'
 tasktracker_edu_validation = 'jba.src.processing.tasktracker_edu_validation:main'
+tasktracker_platform_step_mapping = 'jba.src.processing.tasktracker_platform_step_mapping:main'
 process_course_data = 'jba.src.processing.data_processing:main'
 plot_task_attempt = 'jba.src.plots.task_attempt:main'
 plot_task_solving = 'jba.src.plots.task_solving:main'