From 6201e35e99f3aaa60a9ceae9a251345dddea8f98 Mon Sep 17 00:00:00 2001
From: coolneng <akasroua@gmail.com>
Date: Tue, 11 May 2021 20:41:54 +0200
Subject: [PATCH] Document the data parsing function

---
 src/model.py | 7 ++++++-
 1 file changed, 6 insertions(+), 1 deletion(-)

diff --git a/src/model.py b/src/model.py
index 9a0aef8..5b31d86 100644
--- a/src/model.py
+++ b/src/model.py
@@ -1,6 +1,11 @@
 from tensorflow_io import genome
+from tensorflow import Tensor
+from typing import Tuple
 
 
-def parse_data(filepath):
+def parse_data(filepath) -> Tuple[Tensor, Tensor]:
+    """
+    Reads a FASTQ file and returns the sequences and the qualities
+    """
     HVR = genome.read_fastq(filename=filepath)
     return HVR.sequences, HVR.raw_quality