Code style from styler

NEWS.md

@@ -2,7 +2,7 @@
 
 * Updates to documentation (#109) thanks to Emil Hvitfeldt.
 * Add new tokenizers for tweets, Penn Treebank to `unnest_tokens()`.
-* Better error message (#111).
+* Better error message (#111) and code styling.
 
 # tidytext 0.1.8
 

R/bind_tf_idf.R

@@ -37,7 +37,7 @@
 #'   bind_tf_idf(word, book, n) %>%
 #'   arrange(desc(tf_idf))
 #'
-#'@export
+#' @export
 
 bind_tf_idf <- function(tbl, term, document, n) {
   UseMethod("bind_tf_idf")
@@ -54,7 +54,6 @@ bind_tf_idf.default <- function(tbl, term, document, n) {
 
 #' @export
 bind_tf_idf.data.frame <- function(tbl, term, document, n) {
-
   term <- quo_name(enquo(term))
   document <- quo_name(enquo(document))
   n_col <- quo_name(enquo(n))
@@ -85,5 +84,5 @@ bind_tf_idf_.data.frame <- function(tbl, term, document, n) {
   term <- compat_lazy(term, caller_env())
   document <- compat_lazy(document, caller_env())
   n <- compat_lazy(n, caller_env())
-  bind_tf_idf(tbl, !! term, !! document, !! n)
+  bind_tf_idf(tbl, !!term, !!document, !!n)
 }

R/corpus_tidiers.R

@@ -52,7 +52,7 @@ tidy.Corpus <- function(x, collapse = "\n", ...) {
       # keep as a list column
       return(m)
     }
-    m <- purrr::map_at(m, which(lengths == 0), ~NA)
+    m <- purrr::map_at(m, which(lengths == 0), ~ NA)
 
     ret <- unname(do.call(c, m))
     ## tbl_df() doesn't support POSIXlt format
@@ -115,9 +115,11 @@ tidy.Corpus <- function(x, collapse = "\n", ...) {
 #'
 #' @export
 tidy.corpus <- function(x, ...) {
-  dplyr::tbl_df(data.frame(text = quanteda::texts(x),
-                           quanteda::docvars(x),
-                           stringsAsFactors = FALSE))
+  dplyr::tbl_df(data.frame(
+    text = quanteda::texts(x),
+    quanteda::docvars(x),
+    stringsAsFactors = FALSE
+  ))
 }
 
 
@@ -127,7 +129,7 @@ glance.corpus <- function(x, ...) {
   md <- purrr::compact(quanteda::metacorpus(x))
 
   # turn vectors into list columns
-  md <- purrr::map_if(md, ~length(.) > 1, list)
+  md <- purrr::map_if(md, ~ length(.) > 1, list)
 
   as_data_frame(md)
 }

R/dictionary_tidiers.R

@@ -13,7 +13,8 @@
 #' @export
 tidy.dictionary2 <- function(x, regex = FALSE, ...) {
   ret <- purrr::map_df(x, function(e) data_frame(word = e),
-                       .id = "category") %>%
+                       .id = "category"
+  ) %>%
     mutate(category = as.character(category))
 
   if (regex) {

R/globals.R

@@ -1,4 +1,6 @@
-globalVariables(c("X1", "X2", "X3", "word", "code", "category", "texts",
-                  "i", "j", "x", "Var1", "Var2", "value",
-                  ".ndocs", ".nterm", ".document_total",
-                  "tf", "idf", "sentiments", "document", "term"))
+globalVariables(c(
+  "X1", "X2", "X3", "word", "code", "category", "texts",
+  "i", "j", "x", "Var1", "Var2", "value",
+  ".ndocs", ".nterm", ".document_total",
+  "tf", "idf", "sentiments", "document", "term"
+))

R/lda_tidiers.R

@@ -134,8 +134,10 @@ augment.LDA <- function(x, data, ...) {
       data <- tidy(data)
     } else if (!inherits(data, "data.frame") &&
                !(all(c("document", "term") %in% colnames(data)))) {
-      stop("data argument must either be a simple_triplet_matrix (such as ",
-           "a DocumentTermMatrix) or a table with document and term columns")
+      stop(
+        "data argument must either be a simple_triplet_matrix (such as ",
+        "a DocumentTermMatrix) or a table with document and term columns"
+      )
     }
     ret <- left_join(data, ret, by = c("document", "term"))
   }

R/sparse_casters.R

@@ -65,8 +65,10 @@ cast_sparse <- function(data, row, column, value, ...) {
     j <- match(col_names, col_u)
   }
 
-  ret <- Matrix::sparseMatrix(i = i, j = j, x = values,
-                              dimnames = list(row_u, col_u), ...)
+  ret <- Matrix::sparseMatrix(
+    i = i, j = j, x = values,
+    dimnames = list(row_u, col_u), ...
+  )
 
   ret
 }
@@ -78,10 +80,10 @@ cast_sparse_ <- function(data, row, column, value) {
   row <- compat_lazy(row, caller_env())
   column <- compat_lazy(column, caller_env())
   if (missing(value)) {
-    cast_sparse(data, !! row, !! column)
+    cast_sparse(data, !!row, !!column)
   } else {
     value_col <- compat_lazy(value, caller_env())
-    cast_sparse(data, !! row, !! column, !! value_col)
+    cast_sparse(data, !!row, !!column, !!value_col)
   }
 }
 
@@ -114,7 +116,7 @@ cast_tdm <- function(data, term, document, value,
   term <- quo_name(enquo(term))
   document <- quo_name(enquo(document))
   value <- quo_name(enquo(value))
-  m <- cast_sparse(data, !! term, !! document, !! value, ...)
+  m <- cast_sparse(data, !!term, !!document, !!value, ...)
   tm::as.TermDocumentMatrix(m, weighting = weighting)
 }
 
@@ -127,20 +129,21 @@ cast_tdm_ <- function(data, term, document, value,
   term <- compat_lazy(term, caller_env())
   document <- compat_lazy(document, caller_env())
   value <- compat_lazy(value, caller_env())
-  cast_tdm(data, !! term, !! document, !! value,
-           weighting = weighting, ...)
+  cast_tdm(data, !!term, !!document, !!value,
+           weighting = weighting, ...
+  )
 }
 
 
 
 #' @rdname document_term_casters
 #' @export
 cast_dtm <- function(data, document, term, value,
-                      weighting = tm::weightTf, ...) {
+                     weighting = tm::weightTf, ...) {
   document <- quo_name(enquo(document))
   term <- quo_name(enquo(term))
   value <- quo_name(enquo(value))
-  m <- cast_sparse(data, !! document, !! term, !! value, ...)
+  m <- cast_sparse(data, !!document, !!term, !!value, ...)
   tm::as.DocumentTermMatrix(m, weighting = weighting)
 }
 
@@ -149,12 +152,13 @@ cast_dtm <- function(data, document, term, value,
 #' @inheritParams cast_dtm
 #' @export
 cast_dtm_ <- function(data, document, term, value,
-                     weighting = tm::weightTf, ...) {
+                      weighting = tm::weightTf, ...) {
   document <- compat_lazy(document, caller_env())
   term <- compat_lazy(term, caller_env())
   value <- compat_lazy(value, caller_env())
-  cast_dtm(data, !! document, !! term, !! value,
-           weighting = weighting, ...)
+  cast_dtm(data, !!document, !!term, !!value,
+           weighting = weighting, ...
+  )
 }
 
 
@@ -164,7 +168,7 @@ cast_dfm <- function(data, document, term, value, ...) {
   document <- quo_name(enquo(document))
   term <- quo_name(enquo(term))
   value <- quo_name(enquo(value))
-  m <- cast_sparse(data, !! document, !! term, !! value, ...)
+  m <- cast_sparse(data, !!document, !!term, !!value, ...)
   quanteda::as.dfm(m)
 }
 
@@ -176,5 +180,5 @@ cast_dfm_ <- function(data, document, term, value, ...) {
   document <- compat_lazy(document, caller_env())
   term <- compat_lazy(term, caller_env())
   value <- compat_lazy(value, caller_env())
-  cast_dfm(data, !! document, !! term, !! value)
+  cast_dfm(data, !!document, !!term, !!value)
 }

R/stm_tidiers.R

@@ -120,7 +120,6 @@ tidy.STM <- function(x, matrix = c("beta", "gamma", "theta"), log = FALSE,
 #'
 #' @export
 augment.STM <- function(x, data, ...) {
-
   if (missing(data)) {
     stop("data argument must be provided in order to augment a stm model")
   }
@@ -134,8 +133,10 @@ augment.STM <- function(x, data, ...) {
     mat <- data
     data <- tidy(mat)
   } else {
-    stop("data argument must either be a dfm ",
-         "(from quanteda) or a table with document and term columns")
+    stop(
+      "data argument must either be a dfm ",
+      "(from quanteda) or a table with document and term columns"
+    )
   }
 
   beta <- t(as.matrix(x$beta$logbeta[[1]]))
@@ -167,11 +168,13 @@ augment.STM <- function(x, data, ...) {
 #' @export
 
 glance.STM <- function(x, ...) {
-  ret <- data_frame(k = as.integer(x$settings$dim$K),
-                    docs = x$settings$dim$N,
-                    terms = x$settings$dim$V,
-                    iter = length(x$convergence$bound),
-                    alpha = x$settings$init$alpha)
+  ret <- data_frame(
+    k = as.integer(x$settings$dim$K),
+    docs = x$settings$dim$N,
+    terms = x$settings$dim$V,
+    iter = length(x$convergence$bound),
+    alpha = x$settings$init$alpha
+  )
 
   ret
 }

R/stop_words.R

@@ -45,6 +45,8 @@
 #' @export
 #'
 get_stopwords <- function(language = "en", source = "snowball") {
-  data_frame(word = stopwords::stopwords(language = language, source = source),
-             lexicon = source)
+  data_frame(
+    word = stopwords::stopwords(language = language, source = source),
+    lexicon = source
+  )
 }