Pylint corrections

src/drawing.py

@@ -28,6 +28,7 @@ def draw_number(number, cond_gan):
 
     return generated_image
 
+
 def draw_image(description, cond_gan):
     noise = np.random.normal(size=(1, utils.latent_dim))
     label = keras.utils.to_categorical([description], builders.num_classes)
@@ -47,4 +48,4 @@ def draw_image(description, cond_gan):
     filename = f"./data/images/drawn_image_{description}.png"
     imageio.imwrite(filename, generated_image)
 
-    return generated_image
+    return generated_image

src/nums/cgan.py

@@ -13,7 +13,7 @@ def __init__(self, discriminator, generator, latent_dim, image_size, num_classes
         self.num_classes = num_classes
 
 
-# pylint: disable=too-many-ancestors, too-many-instance-attributes
+# pylint: disable=too-many-ancestors, too-many-instance-attributes, too-many-locals
 class ConditionalGAN(keras.Model):
     def __init__(
         self, config, d_optimizer, g_optimizer, loss_fn

src/nums/training.py

@@ -1,7 +1,8 @@
 from src import utils
+from src.nums import builders
 
-dataset = utils.load_dataset()
-generator, discriminator = utils.build_models()
-cond_gan = utils.build_conditional_gan(generator, discriminator)
+dataset = utils.load_dataset("mnist")
+generator, discriminator = builders.build_models()
+cond_gan = builders.build_conditional_gan(generator, discriminator)
 utils.train_model(dataset, cond_gan)
 utils.save_model_weights(cond_gan, "cond_weights.weights.h5")