modified init.py for new networks

palVJ · Jul 20, 2021 · 3ce66a6 · 3ce66a6
1 parent 229ce17
commit 3ce66a6
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/nilmtk_contrib/disaggregate/__init__.py b/nilmtk_contrib/disaggregate/__init__.py
@@ -6,4  6,9 @@
 from .seq2point import Seq2Point
 from .seq2seq import Seq2Seq
 from .WindowGRU import WindowGRU
-from .rnn import RNN
 from .rnn import RNN
 from .rnn_attention import RNN_attention
 from .rnn_attention_classification import RNN_attention_classification
 from .resnet import ResNet
 from .resnet_classification import ResNet_classification
 from .bert import BERT