linting, moved env, updated contrib credit

This commit is contained in:
Shannon Sands 2025-05-26 14:35:16 +10:00
parent 81d1ebeaef
commit bf12e7df15
83 changed files with 1560 additions and 640 deletions

View file

@ -0,0 +1,24 @@
from atropos.environments.optimizer.wrapper import score_optimizer
from atroposlib.envs.base import BaseEnv, BaseEnvConfig
class OptimizerBenchmarkEnvConfig(BaseEnvConfig):
architecture: str = "mnist"
class OptimizerBenchmarkEnvironment(BaseEnv):
env_config_cls = OptimizerBenchmarkEnvConfig
def __init__(
self,
config: OptimizerBenchmarkEnvConfig,
server_configs=None,
slurm=False,
testing=False,
):
super().__init__(config, server_configs, slurm, testing)
self.architecture = config.architecture
async def evaluate(self, optimizer_code: str, *args, **kwargs):
# This method is required by BaseEnv
return score_optimizer(optimizer_code, self.architecture)