Fix Pandas SettingWithCopyWarning by using loc

Signed-off-by: Alain Anghelidi <alainanghelidi@gmail.com>
kedro-org · Oct 28, 2023 · b790049 · b790049
1 parent 5d24570
commit b790049
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/demo-project/src/demo_project/pipelines/reporting/nodes.py b/demo-project/src/demo_project/pipelines/reporting/nodes.py
@@ -57,7 +57,7 @@ def make_price_histogram(model_input_data: pd.DataFrame) -> go.Figure:
     Returns:
         BaseFigure: Plotly object which is serialised as JSON for rendering
     """
-    price_data_df = model_input_data[["price", "engine_type"]]
+    price_data_df = model_input_data.loc[:, ["price", "engine_type"]]
     p = np.random.dirichlet([1, 1, 1])
     price_data_df["engine_type"] = np.random.choice(
         ["Quantum", "Plasma", "Nuclear"], len(price_data_df), p=p