def get_attr_batched()

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


            def get_attr_batched(output_name, *, score_fn=default_score_fn):
                return batched_get(
                    trajectory_observations,
                    batch_size,
                    lambda minibatch: get_attr_minibatch(
                        minibatch, output_name, score_fn=score_fn
                    ),
                )