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