"""
states/__init__.py - State Machine Initialization
"""
from states.State import State, STATE_REGISTRY
from states.IdleState import IdleState
from states.WalkingState import WalkingState, WalkingFourState

# Register available state instances
STATE_REGISTRY["idle"] = IdleState()
STATE_REGISTRY["walking"] = WalkingState()
STATE_REGISTRY["walking_four"] = WalkingFourState()

__all__ = ["State", "STATE_REGISTRY", "IdleState", "WalkingState", "WalkingFourState"]