add main

ntumslab · Dec 14, 2017 · fb875a3 · fb875a3
1 parent 269f9b7
commit fb875a3
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/src/main.py b/src/main.py
@@ -73,5 +73,6 @@ def main(args):
     np.savetxt(args.output, embeddings, delimiter=',')
 
 
-args = parse_args()
-main(args)
+if __name__ == "__main__":
+    args = parse_args()
+    main(args)