01287d53f5b22f7da20a7b08a836d3a7f6c7af9f,torch_geometric/datasets/s3dis.py,S3DIS,download,#S3DIS#,49

Before Change

        return "data.pt"

    def download(self):
        raise RuntimeError(
            "Dataset not found. Please download {} from {} and move it to {}".
            format(self.raw_file_names, self.url, self.raw_dir))

    def process(self):
        scipy.io.loadmat(self.raw_paths[0])
        raise NotImplementedError

After Change

        extract_zip(path, self.root)
        os.unlink(path)
        shutil.rmtree(self.raw_dir)
        name = self.url.split(os.sep)[-1].split(".")[0]
        os.rename(osp.join(self.root, name), self.raw_dir)

    def process(self):
        with open(self.raw_paths[0], "r") as f:

Instances