quantling · shanshanxu33 · Oct 19, 2024 · Oct 19, 2024 · Oct 19, 2024 · Oct 19, 2024
diff --git a/src/cholesky.jl b/src/cholesky.jl
@@ -406,4 +406,4 @@ function format_matrix(
             return Array(M)
         end
     end
-end
+end
diff --git a/src/input.jl b/src/input.jl
@@ -212,7 +212,7 @@ function loading_data_careful_split(
     n_grams_target_col::Union{Symbol, String} = :Word,
     n_grams_tokenized::Bool = false,
     n_grams_sep_token::Union{Nothing, String} = nothing,
-    grams::Int = 3,
+    grams::Union{Vector{Int64}, Int64}=[3],
     n_grams_keep_sep::Bool = false,
     start_end_token::String = "#",
     random_seed::Int = 314,
-Original file line number
+Diff line change
@@ Expand Up / @@ -406,4 +406,4 @@ function format_matrix( @@
                 return Array(M)
             end
         end
-    end
+    end