Clean up base.py

combo/models/base.py

@@ -5,10 +5,7 @@ import torch.nn as nn
 import utils
 import combo.models.combo_nn as combo_nn
 import combo.utils.checks as checks
-
-
-class Model:
-    pass
+from combo import data
 
 
 class Predictor(nn.Module):
@@ -21,7 +18,6 @@ class Predictor(nn.Module):
 
 
 class Linear(nn.Linear):
-
     def __init__(self,
                  in_features: int,
                  out_features: int,
@@ -147,7 +143,6 @@ class FeedForward(torch.nn.Module):
         return output, feature_maps
 
 
-
 class FeedForwardPredictor(Predictor):
     """Feedforward predictor. Should be used on top of Seq2Seq encoder."""
 
@@ -216,4 +211,3 @@ class FeedForwardPredictor(Predictor):
             hidden_dims=hidden_dims,
             activations=activations,
             dropout=dropout))
-