yapf

ueser · Feb 14, 2018 · 0f374ef · 0f374ef
1 parent db69e76
commit 0f374ef
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 4 deletions.
diff --git a/deepchem/models/tensorgraph/robust_multitask.py b/deepchem/models/tensorgraph/robust_multitask.py
@@ -15,7 +15,6 @@
 
 
 class RobustMultitaskClassifier(TensorGraph):
-
   """Implements a neural network for robust multitasking.
   
   Key idea is to have bypass layers that feed directly from features to task
@@ -371,4 +370,4 @@ def default_generator(self,
           feed_dict[self.features[0]] = X_b
         if w_b is not None and not predict:
           feed_dict[self.task_weights[0]] = w_b
-        yield feed_dict
+        yield feed_dict
diff --git a/deepchem/models/tests/test_overfit.py b/deepchem/models/tests/test_overfit.py
@@ -488,7 +488,6 @@ def test_multitask_regression_overfit(self):
     scores = model.evaluate(dataset, [regression_metric])
     assert scores[regression_metric.name] < .1
 
-
   def test_tf_robust_multitask_regression_overfit(self):
     """Test tf robust multitask overfits tiny data."""
     np.random.seed(123)
@@ -525,7 +524,6 @@ def test_tf_robust_multitask_regression_overfit(self):
     # Eval model on train
     scores = model.evaluate(dataset, [regression_metric])
     assert scores[regression_metric.name] < .2
-
 
   def test_tensorgraph_DTNN_multitask_regression_overfit(self):
     """Test deep tensor neural net overfits tiny data."""