We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent df63f0c commit 551bb33Copy full SHA for 551bb33
opensea_dataset/opensea_dataset.py
@@ -1,6 +1,7 @@
1
import tensorflow_datasets as tfds
2
import os
3
import random
4
+from pathlib import Path
5
6
_DESCRIPTION = "celeblocal data loader"
7
_CITATION = "TFDS Celeb_A"
@@ -27,7 +28,7 @@ def _info(self) -> tfds.core.DatasetInfo:
27
28
29
def _split_generators(self, dl_manager: tfds.download.DownloadManager):
30
# path = dl_manager.download_and_extract('https://s3.amazonaws.com/pytorch-tutorial-assets/img_align_celeba.zip')
- path = os.path.dirname(__file__)
31
+ path = Path(os.path.dirname(__file__))
32
return {
33
'train': self._generate_examples(path),
34
}
0 commit comments