From 5ac81c049faab888b1639525605016503ac18a04 Mon Sep 17 00:00:00 2001
From: coolneng <akasroua@gmail.com>
Date: Tue, 1 Jun 2021 18:34:29 +0200
Subject: [PATCH] Change BASES constant to a local variable

---
 src/model.py         | 6 ++----
 src/preprocessing.py | 2 --
 2 files changed, 2 insertions(+), 6 deletions(-)

diff --git a/src/model.py b/src/model.py
index de6e87c..dbf3bbe 100644
--- a/src/model.py
+++ b/src/model.py
@@ -1,10 +1,8 @@
 from tensorflow.keras import Model, Sequential, layers
 from tensorflow.keras.regularizers import l2
 
-from preprocessing import BASES
 
-
-def build_model(hyper_parameters) -> Model:
+def build_model(hyper_parameters, bases="ACGT") -> Model:
     """
     Builds the CNN model
     """
@@ -41,6 +39,6 @@ def build_model(hyper_parameters) -> Model:
             ),
             layers.Dropout(rate=0.3),
             # Output layer with softmax activation
-            layers.Dense(units=len(BASES), activation="softmax"),
+            layers.Dense(units=len(bases), activation="softmax"),
         ]
     )
diff --git a/src/preprocessing.py b/src/preprocessing.py
index f04b6d6..93fa5b6 100644
--- a/src/preprocessing.py
+++ b/src/preprocessing.py
@@ -4,8 +4,6 @@ from numpy.random import random
 from tensorflow.io import TFRecordWriter
 from tensorflow.train import BytesList, Example, Feature, Features, FloatList
 
-BASES = "ACGT"
-
 
 def generate_example(sequence, weight_matrix):
     schema = {