yapf

deepchem · May 7, 2020 · 5efb773 · 5efb773
1 parent 7994066
commit 5efb773
Showing 1 changed file with 4 additions and 5 deletions.
diff --git a/deepchem/models/multitask.py b/deepchem/models/multitask.py
@@ -13,6 +13,7 @@
 
 logger = logging.getLogger(__name__)
 
+
 class SingletaskToMultitask(Model):
   """
   Convenience class to let singletask models be fit on multitask data.
@@ -21,8 +22,7 @@ class SingletaskToMultitask(Model):
   """
 
   def __init__(self, tasks, model_builder, model_dir=None):
-    super(SingletaskToMultitask, self).__init__(
-        self, model_dir=model_dir)
+    super(SingletaskToMultitask, self).__init__(self, model_dir=model_dir)
     self.tasks = tasks
     self.task_model_dirs = {}
     self.model_builder = model_builder
@@ -45,9 +45,8 @@ def _create_task_datasets(self, dataset):
       task_data_dirs.append(task_data_dir)
     task_datasets = self._to_singletask(dataset, task_data_dirs)
     for task, task_dataset in zip(self.tasks, task_datasets):
-      logger.info(
-          "Dataset for task %s has shape %s" % (task,
-                                                str(task_dataset.get_shape())))
+      logger.info("Dataset for task %s has shape %s" %
+                  (task, str(task_dataset.get_shape())))
     return task_datasets
 
   @staticmethod