src/energy_data_project/T08_WorldCities.py

@@ -14,9 +14,7 @@ def get_population_of_country(df, country):
     """
     cities = df[df["country"] == country]
     print("cities")
-    pop = cities["population"].sum()
-
-    return pop
+    return cities["population"].sum()
 
 def get_cities_beyond_latitude(df, lat, north=True):
     """
@@ -110,4 +108,4 @@ if __name__ == "__main__":
 
 # funktion
 #  -> raise
-#  -> assert
+#  -> assert