def _get_acqf_options()

in aepsych/generators/base.py [0:0]

• 26 lines of code
• 7 McCabe index (conditional complexity)

    def _get_acqf_options(cls, acqf: AcquisitionFunction, config: Config):
        if acqf is not None:
            acqf_name = acqf.__name__
            default_extra_acqf_args = {
                "beta": 3.98,
                "target": 0.75,
                "objective": None,
            }
            extra_acqf_args = {
                k: config.getobj(
                    acqf_name, k, fallback_type=float, fallback=v, warn=False
                )
                for k, v in default_extra_acqf_args.items()
            }
            acqf_args_expected = signature(acqf).parameters.keys()
            extra_acqf_args = {
                k: v for k, v in extra_acqf_args.items() if k in acqf_args_expected
            }
            if (
                "objective" in extra_acqf_args.keys()
                and extra_acqf_args["objective"] is not None
            ):
                extra_acqf_args["objective"] = extra_acqf_args["objective"]()
        else:
            extra_acqf_args = {}

        return extra_acqf_args