type annotations

mhamilton723 · May 9, 2017 · 593fea4 · 593fea4
1 parent 3fd8011
commit 593fea4
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 5 deletions.
diff --git a/tests/test_delta_transformer.py b/tests/test_delta_transformer.py
@@ -1,5 +1,3 @@
-from numpy.testing import assert_equal
-from sklearn.grid_search import GridSearchCV
 from TimeSeriesRegressor.tseries.delta_transformer import *
 import pandas as pd
 

diff --git a/tseries/double_pipeline.py b/tseries/double_pipeline.py
@@ -8,17 +8,16 @@
 from sklearn.base import BaseEstimator, TransformerMixin, RegressorMixin, clone
 from sklearn.pipeline import Pipeline
 
-
 class DoublePipeline(BaseEstimator):
     """
     Base Class for Time Series Estimators
     """
 
     def __init__(self, x_steps, y_steps):
         self.x_steps = x_steps
-        self.x_pipe_ = None
+        self.x_pipe_ = None  # type: Pipeline
         self.y_steps = y_steps
-        self.y_pipe_ = None
+        self.y_pipe_ = None  # type: Pipeline
 
     # TODO handle parameter setting
     def set_params(self, **params):