Add additional arguments to SavedModelPyTFEagerPolicy.
Add additional arguments to SavedModelPyTFEagerPolicy.

PiperOrigin-RevId: 420155234
Change-Id: Iaf7f2c74045feb7f8a882de65fd145992f21f4de

tensorflow/agents,tensorflow/agents