LightZero/lzero/envs/wrappers/__init__.py

from .action_discretization_env_wrapper import *
from .lightzero_env_wrapper import *