Fix typing

apache · Sep 18, 2023 · ccd88be · ccd88be
1 parent 5316fe6
commit ccd88be
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/sdks/python/apache_beam/dataframe/frames_test.py b/sdks/python/apache_beam/dataframe/frames_test.py
@@ -17,6 +17,7 @@
 import re
 import unittest
 import warnings
+from typing import Dict
 
 import numpy as np
 import pandas as pd
@@ -1648,7 +1649,7 @@ def test_pivot_no_index_provided_on_multiindex(self):
         'describe'))
 
 
-def numeric_only_kwargs_for_pandas_2(agg_type: str) -> dict[str, bool]:
+def numeric_only_kwargs_for_pandas_2(agg_type: str) -> Dict[str, bool]:
   """Get proper arguments for numeric_only.
 
   Behavior for numeric_only in these methods changed in Pandas 2 to default