def instrument(client):
    """
    Auto-instruments code to get nice spans
    """
    from atatus.instrumentation.control import instrument
    instrument()
    try:
        import celery
        from atatus.contrib.celery import register_instrumentation
        register_instrumentation(client)
    except ImportError:
        client.logger.debug("Not instrumenting Celery, couldn't import")
