update type hint using typing

ONSdigital · Dec 10, 2024 · 9d75510 · 9d75510
1 parent 922cb6b
commit 9d75510
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/cons_results/imputation/post_imputation.py b/cons_results/imputation/post_imputation.py
@@ -1,4 +1,5 @@
 import warnings
+from typing import List
 
 import numpy as np
 import pandas as pd
@@ -11,7 +12,7 @@
 # Can the total be imputed? my guess would be no
 
 
-def calculate_totals(df: pd.DataFrame, derive_from: list[int]) -> pd.DataFrame:
+def calculate_totals(df: pd.DataFrame, derive_from: List[int]) -> pd.DataFrame:
 
     """
     Returns the sums of a dataframe in which the first level index is in