[Bug] Errors filtering/selecting data from dataframes in R

[annahadji]

I've found a bug in R cells in wrattler when trying to take sections of dataframes in cases such as below:

Case 1:

cols <- c("scenic_rating", "population_density_area_hectares", "employment_score_rate")
d <- data.frame(column = cols, bin_width = c(0.05, 10, 0.05))
feature <- as.character(d[1,]$column) # we would expect "scenic_rating" to be returned

Case 2:

# here we're trying to filter a dataframe by the value of a particular column
subset_df = df[ which(df$crime_category=='total_count'), ] 

Both of the above cases run normally in R, but return the following error message in wrattler:
Error message "Error in walkCode(a, w) : argument "a" is missing, with no default"

[annahadji]

Alternative solution for Case 1:

cols <- c("scenic_rating", "population_density_area_hectares", "employment_score_rate")
d <- data.frame(column = cols, bin_width = c(0.05, 10, 0.05))
feature <- as.character(d[[1]][1]) # features now gives "scenic_rating"

[annahadji]

Alternative solution(s) for Case 2:

Either do the filtering of columns in a Python cell in wrattler, or
subset_df = subset(df, crime_category=='total_count')

[annahadji]

However, I wouldn't say either of these workarounds are exactly ideal since both of the original cases are common approaches to filtering data in R. Additionally, in particular in case (1) it requires knowing the column order as opposed to just the column name.