__rich_repr__ not honoured for dict keys?

[gshiba]

When a dictionary is rich.printed, the keys do not get the same treatment as values.

Code

import attr
import rich

@attr.define(frozen=True, hash=True)
class A:
    a: int
    def __rich_repr__(self):
        yield "aaa", self.a

@attr.define(frozen=True, hash=True)
class B:
    b:i nt
    def __rich_repr__(self):
        yield "bbb", self.b

d = {A(1): B(2), B(3): A(4)}
rich.print(d)

Actual:

{A(a=1): B(bbb=2), B(b=3): A(aaa=4)}

I expected this:

{A(aaa=1): B(bbb=2), B(bbb=3): A(aaa=4)}

[gshiba]

Interestingly, with the @rich.repr.auto decorator, the custom __rich_repr__ is used:

import attr
import rich
import rich.repr

@rich.repr.auto
@attr.define(frozen=True, hash=True)
class A:
    a: int
    def __rich_repr__(self):
        yield "aaa", self.a

@rich.repr.auto
@attr.define(frozen=True, hash=True)
class B:
    b: int
    def __rich_repr__(self):
        yield "bbb", self.b

d = {A(1): B(2), B(3): A(4)}
rich.print(d)

Output

{A(aaa=1): B(bbb=2), B(bbb=3): A(aaa=4)}