code: def restore_schema(task, **kwargs):
      
      from .compat import get_public_schema_name
  
      schema_name = get_public_schema_name()
      include_public = True
  
      if hasattr(task, '_old_schema'):
          schema_name, include_public = task._old_schema
  
      # If the schema names match, don't do anything.
      if connection.schema_name == schema_name:
          return
  
      connection.set_schema(schema_name, include_public=include_public)
docs: Switches the schema back to the one from before running the task.