src/ir/ir.py

import dotenv
from omegaconf import DictConfig, OmegaConf
from hydra import compose, initialize


# load environment variables from `.env` file if it exists
# recursively searches for `.env` in all folders starting from work dir
dotenv.load_dotenv(override=True)

# @hydra.main(version_base = None, config_path="configs/", config_name="test.yaml")
def handle_ir(img_path, input, output):
    with initialize(version_base=None, config_path="configs", job_name="ir"):
        config = compose(config_name="test")
        # print(OmegaConf.to_yaml(cfg))

    # Imports can be nested inside @hydra.main to optimize tab completion
    # https://github.com/facebookresearch/hydra/issues/934
    from src.ir.src import utils
    from src.ir.src.testing_pipeline import test

    # Applies optional utilities
    utils.extras(config)

    # Evaluate model
    return test(config, input, output, img_path)