diff --git a/src/meds/schema.py b/src/meds/schema.py index f6269a3..800add0 100644 --- a/src/meds/schema.py +++ b/src/meds/schema.py @@ -100,7 +100,7 @@ def data_schema(custom_properties=[]): # The subject split schema. -subject_splits_filepath = "subject_splits.parquet" +subject_splits_filepath = os.path.join("metadata", "subject_splits.parquet") train_split = "train" # For ML training. tuning_split = "tuning" # For ML hyperparameter tuning. Also often called "validation" or "dev". @@ -118,7 +118,7 @@ def data_schema(custom_properties=[]): # The dataset metadata schema. # This is a JSON schema. -dataset_metadata_filepath = "dataset.json" +dataset_metadata_filepath = os.path.join("metadata", "dataset.json") dataset_metadata_schema = { "type": "object", @@ -152,7 +152,7 @@ def data_schema(custom_properties=[]): # The code metadata schema. # This is a parquet schema. -code_metadata_filepath = "codes.parquet" +code_metadata_filepath = os.path.join("metadata", "codes.parquet") description_field = "description" description_dtype = pa.string()