Reapply AZ Python fixes

[shreyasvinaya]

Reapplies the AZ changes from #196 after the revert in #197.

[riya-singh28]

@shreyasvinaya I have put some comments after the first review.

[riya-singh28]

@shreyasvinaya Please shift all the fixtures to conftest.py file.

[riya-singh28]

@shreyasvinaya Is a function required here instead of just try-except at the beginning of the file?

[riya-singh28]

@shreyasvinaya Why is the following try-except block used in run_az, but not in run_az_with_img function?

try:

        config_path = f"{AZ_MODELS_PATH}/{az_model}/config.yml"

        with open(config_path, "r") as _:

            config_filename = config_path

    except FileNotFoundError:

        logger.warning("AZ config not found, trying fallback", path=config_path)

        try:

            with open(AZ_MODEL_CONFIG_PATH, "r") as _:

                config_filename = AZ_MODEL_CONFIG_PATH

        except FileNotFoundError:

            raise FileNotFoundError(

                f"AZ_MODEL_CONFIG_PATH not found at {AZ_MODEL_CONFIG_PATH}"

            )

[shreyasvinaya]

handled with the common _resolve_config function

[riya-singh28]

@shreyasvinaya I compared run_az function with the run_az_with_img, and mostly it looks the same except for the image computation and output sizes. In general, this may cause issues in the future if someone updates, given that any change made in both functions.

[shreyasvinaya]

Update: since the core for both are the same, there is now a private function that does the common parts, each of the individual functions now format the output of the private function in their respective formats