openenv/atari_env

1# Copyright (c) Meta Platforms, Inc. and affiliates.2# All rights reserved.3#4# This source code is licensed under the BSD-style license found in the5# LICENSE file in the root directory of this source tree.6 7"""Rubrics for reward computation.8 9See RFC 004 for full design: rfcs/004-rubrics.md10"""11 12from openenv.core.rubrics.base import Rubric13from openenv.core.rubrics.containers import (14    Gate,15    RubricDict,16    RubricList,17    Sequential,18    WeightedSum,19)20from openenv.core.rubrics.llm_judge import LLMJudge21from openenv.core.rubrics.trajectory import (22    ExponentialDiscountingTrajectoryRubric,23    TrajectoryRubric,24)25 26__all__ = [27    # Base28    "Rubric",29    # Containers30    "Sequential",31    "Gate",32    "WeightedSum",33    "RubricList",34    "RubricDict",35    # Trajectory36    "TrajectoryRubric",37    "ExponentialDiscountingTrajectoryRubric",38    # LLM Judge39    "LLMJudge",40]41