def _script_source()

in eland/ml/ltr/feature_logger.py [0:0]

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

    def _script_source(self) -> str:
        query_extractors = self._model_config.query_feature_extractors
        queries = [
            self._to_named_query(extractor.query, extractor.feature_name)
            for extractor in query_extractors
        ]

        return (
            json.dumps(
                {
                    "query": {
                        "bool": {
                            "should": queries,
                            "filter": {"ids": {"values": "##DOC_IDS_JSON##"}},
                        }
                    },
                    "size": "##DOC_IDS_SIZE##",
                    "_source": False,
                }
            )
            .replace('"##DOC_IDS_JSON##"', "{{#toJson}}__doc_ids{{/toJson}}")
            .replace('"##DOC_IDS_SIZE##"', "{{__size}}")
        )