feat: making scripts run via their shortcut commands

deepskies · Oct 22, 2024 · 49bd646 · 49bd646
1 parent 0634e13
commit 49bd646
Show file tree

Hide file tree

Showing 3 changed files with 10 additions and 2 deletions.
diff --git a/__init__.py b/__init__.py
diff --git a/src/scripts/DeepEnsemble.py b/src/scripts/DeepEnsemble.py
@@ -370,7 +370,7 @@ def beta_type(value):
         )
 
 
-if __name__ == "__main__":
+def main():
     """Main execution script for the DeepEnsemble pipeline.
 
     This script performs the following steps:
@@ -555,3 +555,7 @@ def beta_type(value):
         size_df=size_df,
         verbose=verbose,
     )
+
+
+if __name__ == "__main__":
+    main()
diff --git a/src/scripts/DeepEvidentialRegression.py b/src/scripts/DeepEvidentialRegression.py
@@ -320,7 +320,7 @@ def parse_args():
     return config
 
 
-if __name__ == "__main__":
+def main():
     """Main execution script for the DeepEvidentialRegression pipeline.
 
     This script performs the following steps:
@@ -496,3 +496,7 @@ def parse_args():
         size_df=size_df,
         verbose=config.get_item("model", "verbose", "DER"),
     )
+
+
+if __name__ == "__main__":
+    main()