mirror of
https://github.com/NousResearch/atropos.git
synced 2026-04-22 16:48:57 +00:00
linting, moved env, updated contrib credit
This commit is contained in:
parent
81d1ebeaef
commit
bf12e7df15
83 changed files with 1560 additions and 640 deletions
|
|
@ -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)
|
||||
Loading…
Add table
Add a link
Reference in a new issue