Transaction aware celery abstract task

celery.py

class TransactionAwareTask(Task):
    '''
    Task class which is aware of django db transactions and only executes tasks
    after transaction has been committed
    '''
    abstract = True

    def apply_async(self, *args, **kwargs):
        '''
        Unlike the default task in celery, this task does not return an async
        result
        '''
        transaction.on_commit(
            lambda: super(TransactionAwareTask, self).apply_async(
                *args, **kwargs))

class TransactionAwareTask(Task):
    '''
    Task class which is aware of Django DB transactions and
    only executes tasks after the transaction has been committed
    '''
    
    abstract = True

    def apply_async(self, *args, **kwargs):
        cnx = transaction.get_connection()
        if not cnx.in_atomic_block:
            # https://medium.com/gitux/speed-up-django-transaction-hooks-tests-6de4a558ef96
            return super(TransactionAwareTask, self).apply_async(*args, **kwargs)

        # Unlike the default task in celery, this task does not return an async result
        transaction.on_commit(lambda: super(TransactionAwareTask, self).apply_async(*args, **kwargs))

I would like to contribute with some few code lines, the idea is to make it work with transactional and not transactional tests