Skip to content

Commit 551bb33

Browse files
authored
Update opensea_dataset.py
1 parent df63f0c commit 551bb33

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

opensea_dataset/opensea_dataset.py

+2-1
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,7 @@
11
import tensorflow_datasets as tfds
22
import os
33
import random
4+
from pathlib import Path
45

56
_DESCRIPTION = "celeblocal data loader"
67
_CITATION = "TFDS Celeb_A"
@@ -27,7 +28,7 @@ def _info(self) -> tfds.core.DatasetInfo:
2728

2829
def _split_generators(self, dl_manager: tfds.download.DownloadManager):
2930
# path = dl_manager.download_and_extract('https://s3.amazonaws.com/pytorch-tutorial-assets/img_align_celeba.zip')
30-
path = os.path.dirname(__file__)
31+
path = Path(os.path.dirname(__file__))
3132
return {
3233
'train': self._generate_examples(path),
3334
}

0 commit comments

Comments
 (0)