app/env/environment_a.py

"""Environment A: Pairwise DDI and side-effect identification."""

from __future__ import annotations

from app.env.env_core import PolyGuardEnv


class EnvironmentA(PolyGuardEnv):
    def reset(self, **kwargs):
        kwargs.setdefault("difficulty", "easy")
        kwargs.setdefault("sub_environment", "DDI")
        return super().reset(**kwargs)