Y_df = Y_df[Y_df.unique_id == "FOODS_1_001_CA_1"]

# The problem was that the type was a category. # Add this line

Y_df['unique_id'] = Y_df['unique_id'].astype(str)