File size: 1,150 Bytes
e00c2a1 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 | # Copyright (c) Meta Platforms, Inc. and affiliates.
# All rights reserved.
#
# This source code is licensed under the BSD-style license found in the
# LICENSE file in the root directory of this source tree.
"""Energy & Memory RAM Optimization Environment."""
from .client import EnergyOptimizationEnv
from .models import EnergyOptimizationAction, EnergyOptimizationObservation, Task
from .task_graders import (
TASK_GRADERS,
get_grader,
get_all_graders,
get_grader_metadata,
task_1_basic_ram_reduction_grader,
task_2_energy_optimization_grader,
task_3_balanced_optimization_grader,
task_4_advanced_efficiency_grader,
task_5_expert_optimization_grader,
)
__all__ = [
"EnergyOptimizationAction",
"EnergyOptimizationObservation",
"Task",
"EnergyOptimizationEnv",
"TASK_GRADERS",
"get_grader",
"get_all_graders",
"get_grader_metadata",
"task_1_basic_ram_reduction_grader",
"task_2_energy_optimization_grader",
"task_3_balanced_optimization_grader",
"task_4_advanced_efficiency_grader",
"task_5_expert_optimization_grader",
]
|