Commit 896fcb73 authored by Peter Eastman's avatar Peter Eastman
Browse files

Fixed error on Python 2

parent 7c007cf5
Loading
Loading
Loading
Loading
+2 −1
Original line number Diff line number Diff line
@@ -53,7 +53,8 @@ class SingletaskToMultitask(Model):
            % (task, str(task_dataset.get_shape())), self.verbosity)
    return task_datasets

  def _to_singletask(self, dataset, task_dirs):
  @staticmethod
  def _to_singletask(dataset, task_dirs):
    """Transforms a multitask dataset to a collection of singletask datasets."""
    tasks = dataset.get_task_names()
    assert len(tasks) == len(task_dirs)
+1 −1
Original line number Diff line number Diff line
@@ -88,7 +88,7 @@ class TestSingletasktoMultitaskAPI(TestAPI):
    try:
      for task in range(num_tasks):
        task_dirs.append(tempfile.mkdtemp())
      singletask_datasets = SingletaskToMultitask._to_singletask(None, dataset, task_dirs)
      singletask_datasets = SingletaskToMultitask._to_singletask(dataset, task_dirs)
      for task in range(num_tasks):
        singletask_dataset = singletask_datasets[task]
        X_task, y_task, w_task, ids_task = (singletask_dataset.X, singletask_dataset.y, singletask_dataset.w, singletask_dataset.ids)