From 176f6483d0454dcd853394acb1d73e531b8ae41f Mon Sep 17 00:00:00 2001
From: Joshua Peek <josh@joshpeek.com>
Date: Tue, 19 Jun 2012 15:26:56 -0500
Subject: [PATCH] Ensure token probability is less than 1.0

---
 lib/linguist/classifier.rb | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/lib/linguist/classifier.rb b/lib/linguist/classifier.rb
index 4ab6d191..04433490 100644
--- a/lib/linguist/classifier.rb
+++ b/lib/linguist/classifier.rb
@@ -129,10 +129,10 @@ module Linguist
     #
     # Returns Float between 0.0 and 1.0.
     def token_probability(token, language)
-      if @tokens[language][token] == 0
+      if @tokens[language][token].to_f == 0.0
         1 / @tokens_total.to_f
       else
-        @tokens[language][token].to_f / @languages[language].to_f
+        @tokens[language][token].to_f / @language_tokens[language].to_f
       end
     end