agents

Add a check for matching dtypes with action_spec in the action method of tf_policy.py, as well as a test to ensure the check is working as expected.

Change `TimeStep` documentation to reflect that `step_type` and `reward` may not be scalars.

Make py_metrics configurable via gin.

suite_gym: Fix docstring.

Copybara import of the project: