Possibility to pass the figsize to the wrapper either through a tuple or a float defining the diagonal size.
Possibility to pass the figsize to the wrapper either through a tuple or a float defining the diagonal size.

PiperOrigin-RevId: 338062850
Change-Id: I1c92cd8caae04007de79e928ed47d21439e10d19

deepmind/acme,deepmind/acme