verify that fit object exists before calling transform() by checking …

…for vocab existence
chainsawriot · Feb 29, 2020 · ddfb671 · ddfb671
1 parent 90896b2
commit ddfb671
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/R/model_Collocations.R b/R/model_Collocations.R
@@ -240,6 +240,8 @@ Collocations = R6::R6Class(
       invisible(self$collocation_stat)
     },
     transform = function(it) {
+      if(is.null(private$vocabulary))
+        stop("fit object is missing. Please run fit() method first before transform().")
       # if pointer is invalid - init it
       if(is.null(private$phrases_ptr) || is_invalid_ptr(private$phrases_ptr))
         private$phrases_ptr = create_xptr_unordered_set(private$phrases)