removed ensemble shifts in stitch

calico · Feb 21, 2024 · 88b70e4 · 88b70e4
2 parents 6967b4e + 9be07ed
commit 88b70e4
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 4 deletions.
diff --git a/src/baskerville/scripts/hound_snp.py b/src/baskerville/scripts/hound_snp.py
@@ -77,11 +77,12 @@ def main():
         help="Ensemble prediction shifts [Default: %default]",
     )
     parser.add_option(
-        "--stitch", 
+        "--stitch",
         dest="stitch",
-        default=False, 
+        default=False,
         action="store_true",
-        help="Stitch left and right compensation shift predictions for indels [Default: %default]")
+        help=" [Default: %default]",
+    )
     parser.add_option(
         "--stats",
         dest="snp_stats",

diff --git a/src/baskerville/snps.py b/src/baskerville/snps.py
@@ -228,7 +228,7 @@ def score_snps(params_file, model_file, vcf_file, worker_index, options):
                 )
                 alt_preds = np.repeat(alt_preds, repeats=2, axis=0)
                 alt_preds = list(alt_preds)
-                
+
             # flip reference and alternate
             if snps[si].flipped:
                 rp_snp = np.array(alt_preds)