def _get_feature_types()

in metrics/brier_score/brier_score.py [0:0]

• 31 lines of code
• 2 McCabe index (conditional complexity)

    def _get_feature_types(self):
        if self.config_name == "multilist":
            return [
                datasets.Features(
                    {
                        "references": datasets.Sequence(datasets.Value("float")),
                        "predictions": datasets.Sequence(datasets.Value("float")),
                    }
                ),
                datasets.Features(
                    {
                        "references": datasets.Sequence(datasets.Value("string")),
                        "predictions": datasets.Sequence(datasets.Value("float")),
                    }
                ),
            ]
        else:
            return [
                datasets.Features(
                    {
                        "references": datasets.Value("float"),
                        "predictions": datasets.Value("float"),
                    }
                ),
                datasets.Features(
                    {
                        "references": datasets.Value("string"),
                        "predictions": datasets.Value("float"),
                    }
                ),
            ]