Whitespace variable trimming (#5)

* Add unlist()

* Ensure beginning and ending of variable name is trimmed of whitespace

* Add note

Author: coatless

NEWS.md

@@ -5,6 +5,11 @@
 - Improved graceful errors for `fetch_ucirepo()` and `list_available_datasets()`
   when resources are not found/available. ([#3](https://github.com/coatless-rpkg/ucimlrepo/issues/3),
   thanks Prof. Ripley!)
+  
+## Bug fixes
+
+- Fixed internal subset issue with `fetch_ucirepo()` when metadata from the
+  repository had whitespace characters in the variable names. ([#2](https://github.com/coatless-rpkg/ucimlrepo/issues/2))
 
 # ucimlrepo 0.0.1
 

R/fetch-ucirepo.R

@@ -246,11 +246,11 @@ fetch_ucirepo <- function(name, id) {
       stop('Role must be one of "ID", "Feature", "Target", or "Other"')
     }
 
-    variables_by_role[[variable$role]] <- c(variables_by_role[[variable$role]], variable$name)
+    variables_by_role[[variable$role]] <- c(variables_by_role[[variable$role]], trimws(variable$name))
   }
 
   # Extract dataframes for each variable role
-  ids_df <- if (length(variables_by_role$ID) > 0) df[ , variables_by_role$ID, drop = FALSE] else NULL
+  ids_df <- if (length(variables_by_role$ID) > 0) df[ , unlist(variables_by_role$ID), drop = FALSE] else NULL
   features_df <- if (length(variables_by_role$Feature) > 0) df[ , unlist(variables_by_role$Feature), drop = FALSE] else NULL
   targets_df <- if (length(variables_by_role$Target) > 0) df[ , unlist(variables_by_role$Target), drop = FALSE] else NULL