src/scripts/visualize_dataset.py

import argparse
from src.utils.config_loader import Config,set_seed
from src.utils import config_loader
from src.utils.script_utils import validate_config
import importlib
import random


def visualize_dataset(args):
    config_file_path = args.config_file
    config = Config(config_file_path)

    # validate config
    validate_config(config)

    # set config globally and set seed
    config_loader.config = config
    set_seed(config.seed)

    # now visualize the dataset
    visualize_fn = importlib.import_module(f"src.{config.task}.data.visualize_dataset").visualize
    visualize_fn()

def main():
    parser = argparse.ArgumentParser(description="Prepare dataset based on config yaml file")
    parser.add_argument("config_file",type=str)
    args = parser.parse_args()
    visualize_dataset(args)

if __name__=="__main__":
    main()