sbrl-github/research-rainfallradar
1
0
Fork 0
mirror of https://github.com/sbrl/research-rainfallradar synced 2024-11-25 18:33:01 +00:00
Code Issues Projects Releases Packages Wiki Activity

read_metadata: support file inputs as well as dirs

Browse source
This commit is contained in:
Starbeamrainbowlabs 2022-11-10 19:53:30 +00:00
parent aa7d9b8cf6
commit 0aa2ce19f5
Signed by: sbrl
GPG key ID: 1BE5172E637709C2
1 changed files with 4 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

3
aimodel/src/lib/dataset/read_metadata.py
View file

@ -5,6 +5,9 @@ import json
from ..io.readfile import readfile from ..io.readfile import readfile
def read_metadata(dirpath_dataset): def read_metadata(dirpath_dataset):
if os.path.isfile(dirpath_dataset):
filepath_metadata = os.path.join(os.path.dirname(dirpath_dataset), "metadata.jsonl")
else:
filepath_metadata = os.path.join(dirpath_dataset, "metadata.json") filepath_metadata = os.path.join(dirpath_dataset, "metadata.json")
return json.loads(readfile(filepath_metadata)) return json.loads(readfile(filepath_metadata))