def get_experiment()

in genai-on-vertex-ai/gemini/evals_playbook/utils/evals_playbook.py [0:0]

• 9 lines of code
• 3 McCabe index (conditional complexity)

    def get_experiment(self, experiment_id, task_id: str="", as_dict=False):
        where_keys = {}
        if experiment_id:
            where_keys["experiment_id"] = experiment_id
            if task_id:
                where_keys["task_id"] = task_id
            return self._get_one("experiments", where_keys, as_dict=as_dict)
        else:
            raise Exception(f"Experiment ID is required.")