Handle if None is passed from commandline from main

Author: rzitomer

src/github_analysis/main.py

@@ -80,13 +80,19 @@ def main(args):
     print("Frequency Graph Time:\t" +   str(freqGraphTime - motifTime) +                    "\tseconds")
     print("Total Time:\t\t" +           str(freqGraphTime - startTime) +                    "\tseconds")
 
+
+def none_or_str(value):
+    if value == 'None':
+        return None
+    return value
+
 if __name__ == '__main__':
     parser = argparse.ArgumentParser()
     parser.add_argument("-rp",      "--results_path",   help="The folder to output results of the analysis. e.g. embeddings and plots", default="./results/")
     parser.add_argument("-nw",      "--n_workers",      help="The number of workers to use when running the analysis.", default=8, type=int)
     parser.add_argument("-dp",      "--data_path",      help="The path to the commits.feather file. e.g. /home/user/RStudio-Data-Repository/clean_data/commits_by_org.feather", default="./results/")
     parser.add_argument("-np",      "--n_projects",     help="The number of projects to sample from the dataset.", default=1000, type=int)
-    parser.add_argument("-mc",      "--min_commits",    help="The minimum number of commits for a project to be included in the sample.", default=None, type=int)
+    parser.add_argument("-mc",      "--min_commits",    help="The minimum number of commits for a project to be included in the sample.", default=None, type=none_or_str)
     parser.add_argument("-mcount",  "--min_count",      help="The min_count parameter for the graph2vec model.", default=5, type=int)
     parser.add_argument("-nps",     "--n_personas",     help="The number of personas to extract from each cluster.", default=5, type=int)
     parser.add_argument("-nn",      "--n_neurons",      help="The number of neurons to use for Graph2Vec (project level)", default=128, type=int)