Skip to content

Commit

Permalink
Minor fixes in tests
Browse files Browse the repository at this point in the history
  • Loading branch information
Ayushk4 committed Jul 28, 2019
1 parent cb16418 commit 2467ae2
Show file tree
Hide file tree
Showing 2 changed files with 3 additions and 3 deletions.
3 changes: 2 additions & 1 deletion src/document.jl
Original file line number Diff line number Diff line change
Expand Up @@ -171,7 +171,8 @@ A NGramDocument{AbstractString}
* Title: Untitled Document
* Author: Unknown Author
* Timestamp: Unknown Time
* Snippet: ***SAMPLE TEXT NOT AVAILABLE***```
* Snippet: ***SAMPLE TEXT NOT AVAILABLE***
```
"""
function NGramDocument(txt::AbstractString, dm::DocumentMetadata, n::Integer...=1)
NGramDocument(ngramize(dm.language, tokenize(dm.language, String(txt)), n...), (length(n) == 1) ? Int(first(n)) : Int[n...], dm)
Expand Down
3 changes: 1 addition & 2 deletions test/crf.jl
Original file line number Diff line number Diff line change
@@ -1,5 +1,5 @@
using Flux
using Flux: gradient, LSTM, Dense, reset!
using Flux: gradient, LSTM, Dense, reset!, onehot
using TextAnalysis: score_sequence, forward_score

@testset "crf" begin
Expand Down Expand Up @@ -148,4 +148,3 @@ using TextAnalysis: score_sequence, forward_score
@test crf_param_1 != crf_param_2
end
end

0 comments on commit 2467ae2

Please sign in to comment.