CoolFace
Apppublic

albert-einstein-09/codedark

sourceHugging Facemitupdated 8mo agoView on Hugging Face
3likes
__init__.py39 linesDownload Raw Back to codedark
1"""2CodeDark - OpenEnv-Compatible Data Analytics Environment3 4Multi-turn RL environment for training AI agents on data analytics tasks.5 6Example usage:7    from codedark import CodeDarkEnv8 9    env = CodeDarkEnv("http://localhost:8000")10    obs = env.reset()11    print(f"Task: {obs['question']}")12 13    obs = env.run_python("result = df['y'].mean() * 100")14    obs = env.submit_answer(11.26)15    print(f"Reward: {obs['reward']}")16"""17 18from .client import CodeDarkEnv19from .models import (20    CodeDarkAction,21    CodeDarkObservation,22    CodeDarkState,23    ResetRequest,24    StepRequest,25    HealthResponse,26)27 28__all__ = [29    "CodeDarkEnv",30    "CodeDarkAction",31    "CodeDarkObservation",32    "CodeDarkState",33    "ResetRequest",34    "StepRequest",35    "HealthResponse",36]37 38__version__ = "0.1.0"39