Unverified Commit 2ff2d468 authored by Bharath Ramsundar's avatar Bharath Ramsundar Committed by GitHub
Browse files

Merge pull request #1683 from peastman/move

DiskDataset.move() would not overwrite an existing dataset
parents f61ae8d4 b3b0fcdd
Loading
Loading
Loading
Loading
+2 −0
Original line number Diff line number Diff line
@@ -649,6 +649,8 @@ class DiskDataset(Dataset):

  def move(self, new_data_dir):
    """Moves dataset to new directory."""
    if os.path.isdir(new_data_dir):
      shutil.rmtree(new_data_dir)
    shutil.move(self.data_dir, new_data_dir)
    self.data_dir = new_data_dir