def add_properties_to_gather_run()

in src/responsibleai/rai_analyse/create_score_card.py [0:0]

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

def add_properties_to_gather_run(dashboard_info, rai_info):
    included_tools = {
        RAIToolType.CAUSAL: False,
        RAIToolType.COUNTERFACTUAL: False,
        RAIToolType.ERROR_ANALYSIS: False,
        RAIToolType.EXPLANATION: False,
        RAIToolType.SCORECARD: True,
    }

    _logger.info("Adding properties to the gather run")
    run = Run.get_context()

    run_properties = {
        PropertyKeyValues.RAI_INSIGHTS_TYPE_KEY: "PdfGeneration",
        PropertyKeyValues.RAI_INSIGHTS_DASHBOARD_ID_KEY: dashboard_info[
            DashboardInfo.RAI_INSIGHTS_RUN_ID_KEY
        ],
        PropertyKeyValues.RAI_INSIGHTS_RESPONSIBLEAI_VERSION_KEY: responsibleai_version,
        PropertyKeyValues.RAI_INSIGHTS_MODEL_ID_KEY: dashboard_info[
            DashboardInfo.RAI_INSIGHTS_MODEL_ID_KEY
        ],
        PropertyKeyValues.RAI_INSIGHTS_SCORE_CARD_TITLE_KEY: rai_info["ScoreCardTitle"],
    }

    _logger.info("Appending tool present information")
    for k, v in included_tools.items():
        key = PropertyKeyValues.RAI_INSIGHTS_TOOL_KEY_FORMAT.format(k)
        run_properties[key] = str(v)

    _logger.info("Making service call")
    run.add_properties(run_properties)
    _logger.info("Properties added to score card run")