diff --git a/main.py b/main.py index 93413e8..5b817f2 100644 --- a/main.py +++ b/main.py @@ -45,7 +45,7 @@ def inference(first: pandas.core.series.Series, year_diff = int(first['release_year'] - second['release_year']) FS.set_variable('RELEASE_YEAR', year_diff) - runtime_diff = int(first['runtime'] - second['runtime']) + runtime_diff = int(second['runtime'] - first['runtime']) FS.set_variable('RUNTIME', runtime_diff) if not (np.isnan(first['seasons']) or np.isnan(second['seasons'])): @@ -173,4 +173,4 @@ async def recs(production_id: str, scores.remove(production_id) return { 'id': scores - } \ No newline at end of file + }