Silence some pytype errors.

PiperOrigin-RevId: 470746169
Change-Id: Ife61f2d90b66e4240e4db072e9528dad0ea0c600

Author: rchen152

tf_agents/examples/cql_sac/kumar20/dataset/dataset_utils.py

@@ -98,4 +98,4 @@ def create_episode_dataset(
     dtype = observation_dtype if key == 'states' else np.float32
     dataset[key] = np.asarray(value, dtype)
 
-  return dataset
+  return dataset  # pytype: disable=bad-return-type  # dict-kwargs