From 2b622f3180c6554a5142d5f51a2620480ad67a41 Mon Sep 17 00:00:00 2001 From: Richard Evans Date: Thu, 16 May 2024 13:55:19 -0600 Subject: [PATCH] Updated rpy2 API calls in psid_data_setup.py --- ogusa/psid_data_setup.py | 8 +++++++- 1 file changed, 7 insertions(+), 1 deletion(-) diff --git a/ogusa/psid_data_setup.py b/ogusa/psid_data_setup.py index 5fb87182..4f36b1a7 100644 --- a/ogusa/psid_data_setup.py +++ b/ogusa/psid_data_setup.py @@ -6,9 +6,13 @@ import datetime from linearmodels import PanelOLS from rpy2.robjects import r +from rpy2.robjects import conversion +from rpy2.robjects import default_converter from rpy2.robjects import pandas2ri +from rpy2.robjects.packages import importr from ogusa.constants import PSID_NOMINAL_VARS, PSID_CONSTANT_VARS + pandas2ri.activate() pd.options.mode.chained_assignment = "raise" @@ -34,7 +38,9 @@ def prep_data(data="psid1968to2015.RData"): """ # Read data from R into pandas dataframe r["load"](os.path.join(CURDIR, "..", "data", "PSID", data)) - raw_df = r("psid_df") + raw_r_df = r("psid_df") + with (default_converter + pandas2ri.converter).context(): + raw_df = conversion.get_conversion().rpy2py(raw_r_df) # Create unique identifier for each household # note that will define a new household if head or spouse changes