use errno

1 files changed, 8 insertions, 3 deletions

diff --git a/utils/training/estimate_k_mixture_model.cpp b/utils/training/estimate_k_mixture_model.cpp
index 7a827c9..37a88df 100644
--- a/utils/training/estimate_k_mixture_model.cpp
+++ b/utils/training/estimate_k_mixture_model.cpp
@@ -26,7 +26,6 @@
 void print_help(){
     printf("estimate_k_mixture_model [--bigram-file <FILENAME>]\n");
     printf("                         [--deleted-bigram-file <FILENAME]\n");
-    exit(1);
 }
 
 parameter_t compute_interpolation(KMixtureModelSingleGram * deleted_bigram,
@@ -100,16 +99,22 @@ int main(int argc, char * argv[]){
     while ( i < argc ){
         if ( strcmp("--help", argv[i] ) == 0 ){
             print_help();
+            exit(0);
         } else if ( strcmp("--bigram-file", argv[i]) == 0 ){
-            if ( ++i >= argc )
+            if ( ++i >= argc ) {
                 print_help();
+                exit(EINVAL);
+            }
             bigram_filename = argv[i];
         } else if ( strcmp("--deleted-bigram-file", argv[i]) == 0){
-            if ( ++i >= argc )
+            if ( ++i >= argc ) {
                 print_help();
+                exit(EINVAL);
+            }
             deleted_bigram_filename = argv[i];
         } else{
             print_help();
+            exit(EINVAL);
         }
         ++i;
     }