Small fixes

Author: ogrisel

content/python_files/feature_engineering.py

@@ -247,18 +247,6 @@ def load_electricity_load_data(time, data_source_folder):
     & (pl.col("time") < pl.datetime(2021, 10, 31, hour=10, time_zone="UTC"))
 ).skb.eval().plot.line(x="time:T", y="load_mw:Q")
 
-# %% [markdown]
-#
-# Check that the number of rows matches our expectations based on
-# the number of hours that separate the first and the last dates. We can do
-# that by joining with the time range dataframe and checking that the number of
-# rows stays the same.
-
-# %%
-assert (
-    time.join(electricity, on="time", how="inner").shape[0] == time.shape[0]
-).skb.eval()
-
 # %% [markdown]
 #
 # ## Lagged features
@@ -363,7 +351,7 @@ def iqr(col, *, window_size: int):
         & (pl.col("time") < pl.datetime(2021, 12, 31, time_zone="UTC"))
     ).skb.eval()
 ).transform_fold(
-    [f"weather_temperature_2m_{city_name}" for city_name in city_names],
+    [f"weather_temperature_2m_{city_name}" for city_name in city_names.skb.eval()],
 ).mark_line(
     tooltip=True
 ).encode(