As part of the consolidation effort between wrappers and renderers we have removed ContextAwareTrait and ContextAwareInterface to consolidate around the easier pattern using a separate TypedEntityRenderContext context object to hold arbitrary data.
Uses of getContext and setContext should now use the separate object $this->context->offsetGet() and $this->context->offsetSet(). This is likely to have an impact on any custom variant conditions.

Given that this is now a generic context object for the module, it has been renamed to TypedEntityContext. If you have any reference to the old Drupal\typed_entity\Render\TypedEntityRenderContext you should rename those to Drupal\typed_entity\TypedEntityContext.