<filename>cognibench/envs/__init__.py
from .base import CNBEnv
from .env import BanditEnv, ClassicalConditioningEnv

<filename>cognibench/envs/__init__.py
from .base import CNBEnv
from .env import BanditEnv, ClassicalConditioningEnv
