From 4721f5568c7014c1e75193ad46359797dd859cad Mon Sep 17 00:00:00 2001
From: LucaCappelletti94 <cappelletti.luca94@gmail.com>
Date: Thu, 16 Jul 2020 17:02:43 +0200
Subject: [PATCH] Added property for the nucleotides used in bed sequence

---
 keras_bed_sequence/bed_sequence.py | 8 +++++++-
 1 file changed, 7 insertions(+), 1 deletion(-)

diff --git a/keras_bed_sequence/bed_sequence.py b/keras_bed_sequence/bed_sequence.py
index e6a7cc7..236f60e 100644
--- a/keras_bed_sequence/bed_sequence.py
+++ b/keras_bed_sequence/bed_sequence.py
@@ -54,6 +54,7 @@ def __init__(
 
         self._window_length = (bed.chromEnd - bed.chromStart).values[0]
         self._genome = assembly
+        self._nucleotides = nucleotides
         self._nucleotides_number = len(nucleotides)
         self._unknown_nucleotide_value = unknown_nucleotide_value
 
@@ -62,7 +63,7 @@ def __init__(
         sequences = self._genome.bed_to_sequence(bed).sequence.values.astype(str)
 
         super().__init__(
-            nucleotides_to_numbers(nucleotides, sequences),
+            nucleotides_to_numbers(self.nucleotides, sequences),
             batch_size,
             seed=seed,
             elapsed_epochs=elapsed_epochs
@@ -73,6 +74,11 @@ def window_length(self) -> int:
         """Return number of nucleotides in a window."""
         return self._window_length
 
+    @property
+    def nucleotides(self) -> int:
+        """Return number of nucleotides considered."""
+        return self._nucleotides
+    
     @property
     def nucleotides_number(self) -> int:
         """Return number of nucleotides considered."""