app/env/environment_b.py

"""Environment B: Regimen risk reduction and medication optimization."""

from __future__ import annotations

from app.env.env_core import PolyGuardEnv


class EnvironmentB(PolyGuardEnv):
    def reset(self, **kwargs):
        kwargs.setdefault("difficulty", "medium")
        kwargs.setdefault("sub_environment", "REGIMEN_RISK")
        return super().reset(**kwargs)