Build zipline-ai with Commands "zipline compile" and "zipline run"

api/py/ai/chronon/repo/compile.py

@@ -48,7 +48,7 @@ def get_folder_name_from_class_name(class_name):
     return {v.__name__: k for k, v in FOLDER_NAME_TO_CLASS.items()}[class_name]
 
 
-@click.command()
+@click.command(name='compile')
 @click.option(
     '--chronon_root',
     envvar='CHRONON_ROOT',

api/py/ai/chronon/repo/explore.py

@@ -383,7 +383,8 @@ def load_team_data(path):
     "_events_without_topics": events_without_topics
 }
 
-if __name__ == "__main__":
+
+def main():
     parser = argparse.ArgumentParser(description="Explore tool for chronon")
     parser.add_argument("keyword", help="Keyword to look up keys")
     parser.add_argument("--conf-root", help="Conf root for the configs", default=CWD)
@@ -413,3 +414,7 @@ def load_team_data(path):
     else:
         group_bys = find_in_index(gb_index, args.keyword)
         display_entries(group_bys, args.keyword, root=root, trim_paths=True)
+
+
+if __name__ == "__main__":
+    main()