update to v0.1.5

nextgenusfs · Apr 7, 2016 · fcf4bb9 · fcf4bb9
1 parent 061a808
commit fcf4bb9
Showing 1 changed file with 8 additions and 6 deletions.
diff --git a/bin/funannotate-predict.py b/bin/funannotate-predict.py
@@ -193,14 +193,16 @@ def __init__(self,prog):
     #setup weights file for EVM
     Weights = os.path.join(args.out, 'predict_misc', 'weights.evm.txt')
     with open(Weights, 'w') as output:
+        sources = []
+        with open(Predictions, 'rU') as preds:
+            for line in preds:
+                source = line.split('\t')[1]
+                if source not in sources:
+                    sources.append(source)
         if args.pasa_gff:
             output.write("OTHER_PREDICTION\ttransdecoder\t10\n")
-            output.write("ABINITIO_PREDICTION\taugustus\t1\n")
-            output.write("ABINITIO_PREDICTION\tgenemark\t1\n")
-        else:
-            output.write("ABINITIO_PREDICTION\taugustus\t1\n")
-            output.write("ABINITIO_PREDICTION\tgenemark\t1\n")
-
+        for i in sources:
+            output.write("ABINITIO_PREDICTION\t%s\t1\n" % i)
         output.write("PROTEIN\tprotein2genome\t1\n")
         output.write("TRANSCRIPT\test2genome\t1\n")
     Exonerate = os.path.abspath(Exonerate)