fix(): fix e2e ml tests, handle no entries for vacancies

This commit is contained in:
gitgernit
2025-11-23 11:44:08 +03:00
parent a879da4ed5
commit b15282baef
8 changed files with 40 additions and 27 deletions
@@ -46,7 +46,7 @@ class PredictModelInteractor:
def _predict_salary(self, vacancies: list[VacancyInput], resume_skills: list[str]) -> tuple[Decimal, Decimal]:
if not vacancies:
return Decimal(50000), Decimal(80000)
return Decimal(0), Decimal(0)
vacancy_weights: list[float] = []
for vacancy in vacancies:
@@ -56,7 +56,7 @@ class PredictModelInteractor:
total_weight = sum(vacancy_weights)
if total_weight == 0:
return Decimal(50000), Decimal(80000)
return Decimal(0), Decimal(0)
weighted_from_sum = Decimal(0)
weighted_to_sum = Decimal(0)
@@ -143,6 +143,9 @@ class PredictModelInteractor:
if skill in candidate_skills
}
if not candidate_skills:
return []
frequencies = [skill_frequencies[skill] for skill in candidate_skills]
avg_salaries = [float(skill_avg_salaries[skill]) for skill in candidate_skills]