def generate()

in understanding_rl_vision/rl_clarity/interface.py [0:0]

• 74 lines of code
• 4 McCabe index (conditional complexity)

def generate(
    *,
    output_dir,
    model_bytes,
    observations,
    observations_full=None,
    trajectories,
    policy_logits_name,
    value_function_name,
    env_name=None,
    numpy_precision=6,
    inline_js=True,
    inline_large_json=None,
    batch_size=512,
    action_combos=None,
    action_group_fns=[
        lambda combo: "RIGHT" in combo,
        lambda combo: "LEFT" in combo,
        lambda combo: "UP" in combo,
        lambda combo: "DOWN" in combo,
        lambda combo: "RIGHT" not in combo
        and "LEFT" not in combo
        and "UP" not in combo
        and "DOWN" not in combo,
    ],
    layer_kwargs={},
    input_layer_include=False,
    input_layer_name="input",
    gae_gamma=None,
    gae_lambda=None,
    trajectory_bookmarks=16,
    nmf_features=8,
    nmf_attr_opts=None,
    vis_subdiv_mults=[0.25, 0.5, 1, 2],
    vis_subdiv_mult_default=1,
    vis_expand_mults=[1, 2, 4, 8],
    vis_expand_mult_default=4,
    vis_thumbnail_num_mult=4,
    vis_thumbnail_expand_mult=4,
    scrub_range=(42 / 64, 44 / 64),
    attr_integrate_steps=10,
    attr_max_paths=None,
    attr_policy=False,
    attr_single_channels=True,
    observations_subdir="observations/",
    trajectories_subdir="trajectories/",
    trajectories_scrub_subdir="trajectories_scrub/",
    features_subdir="features/",
    thumbnails_subdir="thumbnails/",
    attribution_subdir="attribution/",
    attribution_scrub_subdir="attribution_scrub/",
    features_grids_subdir="features_grids/",
    attribution_totals_subdir="attribution_totals/",
    video_height="16em",
    video_width="16em",
    video_speed=12,
    policy_display_height="2em",
    policy_display_width="40em",
    navigator_width="24em",
    scrubber_height="4em",
    scrubber_width="48em",
    scrubber_visible_duration=256,
    legend_item_height="6em",
    legend_item_width="6em",
    feature_viewer_height="40em",
    feature_viewer_width="40em",
    attribution_weight=0.9,
    graph_colors={
        "v": "green",
        "action": "red",
        "action_group": "orange",
        "advantage": "blue",
    },
    trajectory_color="blue",