# HG changeset patch
# User galaxytrakr
# Date 1777480435 0
# Node ID 954eccb7cc48600fc0aa11d789d84f66e5037421
# Parent  fe9ff3859d68c9067db4ad196102aab581938074
planemo upload commit 8da69fa6fb55921a1b005486e59563f0c6956c2e

diff -r fe9ff3859d68 -r 954eccb7cc48 predict_source.py
--- a/predict_source.py	Wed Apr 29 15:26:26 2026 +0000
+++ b/predict_source.py	Wed Apr 29 16:33:55 2026 +0000
@@ -10,11 +10,10 @@
 
 def main():
     parser = argparse.ArgumentParser(description="Predict the source of an isolate using a trained Random Forest model and Mash distances.")
+    parser.add_argument('--version', action='version', version='PlasmidTrakr v0.1.0')
+
     parser.add_argument("-i", "--input", required=True, help="Input Mash screen/dist file for one or more isolates.")
-    
-    # --- KEY FIX 1: Replaced -m and -f with a single -b (bundle) argument ---
     parser.add_argument("-b", "--bundle", required=True, help="Path to the bundled model and features (.joblib file)")
-    
     parser.add_argument("-t", "--threshold", type=float, default=0.95, help="Mash identity threshold (default: 0.95)")
     parser.add_argument("-o", "--output", default="predictions.tsv", help="Output file for predictions (default: predictions.tsv)")
     args = parser.parse_args()