diff --git a/learntools/sql/ex5.py b/learntools/sql/ex5.py index 6d3f6e562..3e80ecccd 100644 --- a/learntools/sql/ex5.py +++ b/learntools/sql/ex5.py @@ -97,8 +97,8 @@ def check(self, results): assert (len(results) == len(rides_per_year_answer)), ("The results don't look right. Try again.") # check 3: one value in particular year_to_check = list(rides_per_year_answer["year"])[-1] - correct_number = int(rides_per_year_answer.loc[rides_per_year_answer["year"]==year_to_check]["num_trips"].values) - submitted_number = int(results.loc[results["year"]==year_to_check]["num_trips"].values) + correct_number = rides_per_year_answer.loc[rides_per_year_answer["year"]==year_to_check]["num_trips"].values[0] + submitted_number = results.loc[results["year"]==year_to_check]["num_trips"].values[0] assert (correct_number == submitted_number), ("The results don't look right. Try again.") _hint = "Start your query with `SELECT EXTRACT(YEAR FROM trip_start_timestamp) AS year, COUNT(1) AS num_trips`."