chore: lint

Signed-off-by: Avik Basu <[email protected]>
numaproj · Jun 12, 2024 · b775e1c · b775e1c
1 parent 3a9f926
commit b775e1c
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 3 deletions.
diff --git a/numalogic/models/autoencoder/variants/icvanilla.py b/numalogic/models/autoencoder/variants/icvanilla.py
@@ -125,8 +125,8 @@ def _construct_layers(self, layersizes: Sequence[int]) -> nn.ModuleList:
 
 
 class VanillaICAE(BaseAE):
-    r"""Vanilla Autoencoder model with Independent isolated Channels based on the vanilla encoder and decoder.
-        Each channel is an isolated neural network.
+    r"""Vanilla Autoencoder model with Independent isolated Channels based
+    on the vanilla encoder and decoder. Each channel is an isolated neural network.
 
     Args:
     ----

diff --git a/tests/models/autoencoder/test_trainer.py b/tests/models/autoencoder/test_trainer.py
@@ -13,7 +13,6 @@
 from numalogic.models.autoencoder.variants import (
     Conv1dAE,
     LSTMAE,
-    SparseVanillaAE,
     TransformerAE,
     SparseConv1dAE,
     VanillaAE,