From 8a8c22ca491949a0853e56ab692b19fc0d8f4511 Mon Sep 17 00:00:00 2001
From: valentynbez <walentyn.bzsz@gmail.com>
Date: Mon, 10 Jun 2024 16:49:04 +0200
Subject: [PATCH] fix: compilation issue

---
 mDeepFRI/predict.pyx | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/mDeepFRI/predict.pyx b/mDeepFRI/predict.pyx
index 9b47c5e..1c8967d 100644
--- a/mDeepFRI/predict.pyx
+++ b/mDeepFRI/predict.pyx
@@ -136,7 +136,8 @@ cdef class Predictor(object):
         self,
         seqres: str,
         cmap = None,
-        chain: str = ""
+        chain: str = "",
+        format: str = "tsv",
     ):
         """
         Computes GO/EC predictions for a single protein chain from sequence and contact map.
@@ -145,6 +146,7 @@ cdef class Predictor(object):
             seqres (str): protein sequence.
             cmap (np.array): contact map.
             chain (str): protein ID.
+            format (str): output format. Options: "tsv", "vector".
 
         Returns:
             list: list of GO/EC predictions.