File size: 2,059 Bytes
57edf96 433cefc 5a08768 e8f8c7b 57edf96 433cefc d99151a 433cefc 71b3314 9b145be 57edf96 5a08768 e8f8c7b d99151a 71b3314 9b145be 57edf96 | 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 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 | # 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,
)
from task_registry import (
TASK_REGISTRY,
get_all_tasks_with_graders,
get_task_grader,
get_tasks_count,
is_grader_requirement_met,
)
from grader_manifest import (
GRADERS_MANIFEST,
get_graders_manifest,
get_active_graders_count,
get_grader_names,
is_validator_satisfied,
)
from grader_discovery import (
GRADER_DISCOVERY,
get_grader_discovery,
get_graders_count,
is_requirement_met,
get_all_grader_references,
verify_all_graders_callable,
)
__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",
"TASK_REGISTRY",
"get_all_tasks_with_graders",
"get_task_grader",
"get_tasks_count",
"is_grader_requirement_met",
"GRADERS_MANIFEST",
"get_graders_manifest",
"get_active_graders_count",
"get_grader_names",
"is_validator_satisfied",
"GRADER_DISCOVERY",
"get_grader_discovery",
"get_graders_count",
"is_requirement_met",
"get_all_grader_references",
"verify_all_graders_callable",
]
|