try: from .client import main from .models import Action from .models import EnvironmentState from .models import Observation except ImportError: from client import main from models import Action from models import EnvironmentState from models import Observation __all__ = [ "Action", "Observation", "EnvironmentState", "main", ]