From 0aa2ce19f5d7e58cffc0239e7091e52ef790585b Mon Sep 17 00:00:00 2001 From: Starbeamrainbowlabs Date: Thu, 10 Nov 2022 19:53:30 +0000 Subject: [PATCH] read_metadata: support file inputs as well as dirs --- aimodel/src/lib/dataset/read_metadata.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/aimodel/src/lib/dataset/read_metadata.py b/aimodel/src/lib/dataset/read_metadata.py index 2a1dc5f..03f6941 100644 --- a/aimodel/src/lib/dataset/read_metadata.py +++ b/aimodel/src/lib/dataset/read_metadata.py @@ -5,6 +5,9 @@ import json from ..io.readfile import readfile def read_metadata(dirpath_dataset): - filepath_metadata = os.path.join(dirpath_dataset, "metadata.json") + 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") return json.loads(readfile(filepath_metadata)) \ No newline at end of file