From ccd88be2309b3da26e4f037c7aa4fe963892dc22 Mon Sep 17 00:00:00 2001 From: Christopher Neffshade Date: Mon, 18 Sep 2023 15:37:17 +0000 Subject: [PATCH] Fix typing --- sdks/python/apache_beam/dataframe/frames_test.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/sdks/python/apache_beam/dataframe/frames_test.py b/sdks/python/apache_beam/dataframe/frames_test.py index 7feee9154c95..b8b8cb733ef1 100644 --- 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