You've already forked RekomenciBackend
fix(): fix e2e ml tests, handle no entries for vacancies
This commit is contained in:
@@ -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]
|
||||
|
||||
|
||||
Reference in New Issue
Block a user