Update debug_ee.py

debug_ee.py

@@ -107,11 +107,11 @@ def run_check(original_name, ee_name, seed, prompt="Hello, how are you?"):
     return embed_match and logit_match
 
 
-if __name__ == "__main__":
-    parser = argparse.ArgumentParser()
-    parser.add_argument("--original", required=True)
-    parser.add_argument("--ee",       required=True)
-    parser.add_argument("--seed",     type=int, required=True)
-    parser.add_argument("--prompt",   default="Hello, how are you?")
-    args = parser.parse_args()
-    run_check(args.original, args.ee, args.seed, args.prompt)
+#if __name__ == "__main__":
+parser = argparse.ArgumentParser()
+parser.add_argument("--original", required=True)
+parser.add_argument("--ee",       required=True)
+parser.add_argument("--seed",     type=int, required=True)
+parser.add_argument("--prompt",   default="Hello, how are you?")
+args = parser.parse_args()
+run_check(args.original, args.ee, args.seed, args.prompt)