Improves handling of non-fiction classification data.

[quanlib.git] / main.rb

diff --git a/main.rb b/main.rb
index 6e7a44184ff4b94357e960ef6e5f4c3d6b07a4d1..4ea9c7057de2e456b65eecb2fcad0515b3fbc283 100644 (file)
--- a/main.rb
+++ b/main.rb
@@ -12,6 +12,9 @@ def handleArg(arg)
   if "--purge" == arg
     puts 'Purging database...'
     @store.dropSchema()
+    if File.exists?(Store.unclassified_csv)
+      File.delete(Store.unclassified_csv)
+    end
   elsif arg.start_with?("--")
     abort('ERROR:  Unrecognized option "' + arg + '".')
   end