prevent cli functions from getting fired on import (axolotl-ai-cloud#581

)
mkeoliya · Sep 15, 2023 · d92a4aa · d92a4aa
1 parent 56f7ff0
commit d92a4aa
Show file tree

Hide file tree

Showing 4 changed files with 8 additions and 4 deletions.
diff --git a/src/axolotl/cli/inference.py b/src/axolotl/cli/inference.py
@@ -23,4 +23,5 @@ def do_cli(config: Path = Path("examples/"), **kwargs):
     do_inference(cfg=parsed_cfg, cli_args=parsed_cli_args)
 
 
-fire.Fire(do_cli)
+if __name__ == "__main__":
+    fire.Fire(do_cli)
diff --git a/src/axolotl/cli/merge_lora.py b/src/axolotl/cli/merge_lora.py
@@ -23,4 +23,5 @@ def do_cli(config: Path = Path("examples/"), **kwargs):
     do_merge_lora(cfg=parsed_cfg, cli_args=parsed_cli_args)
 
 
-fire.Fire(do_cli)
+if __name__ == "__main__":
+    fire.Fire(do_cli)
diff --git a/src/axolotl/cli/shard.py b/src/axolotl/cli/shard.py
@@ -38,4 +38,5 @@ def do_cli(config: Path = Path("examples/"), **kwargs):
     shard(cfg=parsed_cfg, cli_args=parsed_cli_args)
 
 
-fire.Fire(do_cli)
+if __name__ == "__main__":
+    fire.Fire(do_cli)
diff --git a/src/axolotl/cli/train.py b/src/axolotl/cli/train.py
@@ -32,4 +32,5 @@ def do_cli(config: Path = Path("examples/"), **kwargs):
     train(cfg=parsed_cfg, cli_args=parsed_cli_args, dataset_meta=dataset_meta)
 
 
-fire.Fire(do_cli)
+if __name__ == "__main__":
+    fire.Fire(do_cli)