unit test fix

owid · Aug 1, 2023 · 5a516ca · 5a516ca
1 parent 4390192
commit 5a516ca
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/etl/steps/data/garden/artificial_intelligence/2023-07-25/cset.py b/etl/steps/data/garden/artificial_intelligence/2023-07-25/cset.py
@@ -43,7 +43,7 @@ def run(dest_dir: str) -> None:
     # List of columns to include for conversion to millions (investment values)
     _investment_cols = [col for col in merged_total.columns if "investment" in col]
     # Convert all other columns to million
-    merged_total[_investment_cols] *= 1e6
+    merged_total.loc[:, _investment_cols] *= 1e6
 
     # Import US CPI data from the API
     df_wdi_cpi_us = us_cpi.import_US_cpi_API()