diff --git a/src/sasctl/pzmm/write_score_code.py b/src/sasctl/pzmm/write_score_code.py index 0bb5dc9c..5162083c 100644 --- a/src/sasctl/pzmm/write_score_code.py +++ b/src/sasctl/pzmm/write_score_code.py @@ -768,12 +768,12 @@ def impute_missing_values(data): """ self.score_code += ( f"\n\n{'':4}# Specify downcasting behavior for pandas 2.x to avoid warnings\n" - + f"{'':4}if int(pd.__version__.split('.')[0]) == 2:\n{'':8}pd.set_option('future.no_silent_downcasting', True)\n" + + f"{'':4}if int(pd.__version__.split('.')[0]) == 2 and int(pd.__version__.split('.')[1]) >= 2:\n{'':8}pd.set_option('future.no_silent_downcasting', True)\n" + f"{'':4}return data.replace(r'^\\s*\\.$', np.nan, regex=True).fillna(impute_values).infer_objects()\n" ) """ # Specify downcasting behavior for pandas 2.x to avoid warnings - if int(pd.__version__.split('.')[0]) == 2: + if int(pd.__version__.split('.')[0]) == 2 and int(pd.__version__.split('.')[1]) >= 2: pd.set_option('future.no_silent_downcasting', True) return data.replace(r'^\s*\.$', np.nan, regex=True).fillna(impute_values).infer_objects() """