20 lines
505 B
Python
20 lines
505 B
Python
|
import os
|
||
|
import typing as t
|
||
|
|
||
|
HERE = os.path.realpath(os.path.dirname(__file__))
|
||
|
|
||
|
|
||
|
def get_data_path(dataset_name: t.Optional[str] = None) -> str:
|
||
|
'''
|
||
|
Returns an absolute path to either the data folder, or a specific dataset if
|
||
|
`dataset_name` is supplied.
|
||
|
'''
|
||
|
if 'WAIFU_DATA_PATH' in os.environ:
|
||
|
return os.environ['WAIFU_DATA_PATH']
|
||
|
|
||
|
components = [HERE, "..", "..", "data"]
|
||
|
if dataset_name:
|
||
|
components.append(dataset_name)
|
||
|
|
||
|
return os.path.join(*components)
|