Path	Lines of Code
tests_nightly_pypi.sh	27
tests_release.sh	168
tf_agents/agents/behavioral_cloning/behavioral_cloning_agent_test.py	368
tf_agents/agents/categorical_dqn/categorical_dqn_agent_test.py	388
tf_agents/agents/cql/cql_sac_agent_test.py	469
tf_agents/agents/data_converter_test.py	268
tf_agents/agents/ddpg/actor_network_test.py	109
tf_agents/agents/ddpg/actor_rnn_network_test.py	74
tf_agents/agents/ddpg/critic_network_test.py	95
tf_agents/agents/ddpg/critic_rnn_network_test.py	122
tf_agents/agents/ddpg/ddpg_agent_test.py	183
tf_agents/agents/dqn/dqn_agent_test.py	417
tf_agents/agents/ppo/ppo_actor_network_test.py	267
tf_agents/agents/ppo/ppo_agent_test.py	1108
tf_agents/agents/ppo/ppo_policy_test.py	372
tf_agents/agents/ppo/ppo_utils_test.py	86
tf_agents/agents/qtopt/qtopt_agent_test.py	216
tf_agents/agents/random/random_agent_test.py	62
tf_agents/agents/reinforce/reinforce_agent_test.py	631
tf_agents/agents/sac/sac_agent_test.py	435
tf_agents/agents/sac/tanh_normal_projection_network_test.py	64
tf_agents/agents/td3/td3_agent_test.py	297
tf_agents/agents/test_util.py	42
tf_agents/agents/tf_agent_test.py	142
tf_agents/bandits/agents/bernoulli_thompson_sampling_agent_test.py	142
tf_agents/bandits/agents/dropout_thompson_sampling_agent_test.py	171
tf_agents/bandits/agents/exp3_agent_test.py	133
tf_agents/bandits/agents/exp3_mixture_agent_test.py	145
tf_agents/bandits/agents/greedy_multi_objective_neural_agent_test.py	474
tf_agents/bandits/agents/greedy_reward_prediction_agent_test.py	517
tf_agents/bandits/agents/linear_bandit_agent_test.py	697
tf_agents/bandits/agents/mixture_agent_test.py	196
tf_agents/bandits/agents/neural_boltzmann_agent_test.py	127
tf_agents/bandits/agents/neural_epsilon_greedy_agent_test.py	103
tf_agents/bandits/agents/neural_linucb_agent_test.py	407
tf_agents/bandits/agents/utils_test.py	117
tf_agents/bandits/environments/bandit_tf_environment_test.py	107
tf_agents/bandits/environments/bernoulli_action_mask_tf_environment_test.py	56
tf_agents/bandits/environments/bernoulli_py_environment_test.py	20
tf_agents/bandits/environments/classification_environment_test.py	162
tf_agents/bandits/environments/dataset_utilities_test.py	23
tf_agents/bandits/environments/drifting_linear_environment_test.py	116
tf_agents/bandits/environments/non_stationary_stochastic_environment_test.py	69
tf_agents/bandits/environments/piecewise_bernoulli_py_environment_test.py	32
tf_agents/bandits/environments/piecewise_stochastic_environment_test.py	112
tf_agents/bandits/environments/random_bandit_environment_test.py	148
tf_agents/bandits/environments/stationary_stochastic_per_arm_py_environment_test.py	87
tf_agents/bandits/environments/stationary_stochastic_py_environment_test.py	163
tf_agents/bandits/environments/stationary_stochastic_structured_py_environment_test.py	52
tf_agents/bandits/environments/wheel_py_environment_test.py	81
tf_agents/bandits/metrics/tf_metrics_test.py	163
tf_agents/bandits/multi_objective/multi_objective_scalarizer_test.py	428
tf_agents/bandits/networks/global_and_arm_feature_network_test.py	174
tf_agents/bandits/networks/heteroscedastic_q_network_test.py	241
tf_agents/bandits/policies/bernoulli_thompson_sampling_policy_test.py	165
tf_agents/bandits/policies/boltzmann_reward_prediction_policy_test.py	403
tf_agents/bandits/policies/categorical_policy_test.py	157
tf_agents/bandits/policies/constraints_test.py	386
tf_agents/bandits/policies/greedy_multi_objective_neural_policy_test.py	509
tf_agents/bandits/policies/greedy_reward_prediction_policy_test.py	409
tf_agents/bandits/policies/linalg_test.py	110
tf_agents/bandits/policies/linear_bandit_policy_test.py	413
tf_agents/bandits/policies/loss_utils_test.py	27
tf_agents/bandits/policies/mixture_policy_test.py	143
tf_agents/bandits/policies/neural_linucb_policy_test.py	487
tf_agents/bandits/policies/policy_utilities_test.py	78
tf_agents/bandits/specs/__init__.py	1
tf_agents/bandits/specs/utils.py	9
tf_agents/benchmark/dqn_benchmark_test.py	218
tf_agents/benchmark/perfzero_benchmark_test.py	51
tf_agents/benchmark/utils_test.py	34
tf_agents/distributions/gumbel_softmax_test.py	32
tf_agents/distributions/masked_test.py	32
tf_agents/distributions/shifted_categorical_test.py	54
tf_agents/distributions/utils_test.py	183
tf_agents/drivers/dynamic_episode_driver_test.py	185
tf_agents/drivers/dynamic_step_driver_test.py	160
tf_agents/drivers/py_driver_test.py	207
tf_agents/drivers/test_utils.py	210
tf_agents/drivers/test_utils_test.py	35
tf_agents/drivers/tf_driver_test.py	200
tf_agents/environments/atari_preprocessing_test.py	70
tf_agents/environments/atari_wrappers_test.py	41
tf_agents/environments/batched_py_environment_test.py	123
tf_agents/environments/dm_control_wrapper_test.py	63
tf_agents/environments/gym_wrapper_test.py	347
tf_agents/environments/parallel_py_environment_test.py	252
tf_agents/environments/py_environment_test.py	38
tf_agents/environments/random_py_environment_test.py	130
tf_agents/environments/random_tf_environment_test.py	64
tf_agents/environments/suite_atari_test.py	42
tf_agents/environments/suite_bsuite_test.py	31
tf_agents/environments/suite_dm_control_test.py	36
tf_agents/environments/suite_gym_test.py	64
tf_agents/environments/suite_mujoco_test.py	38
tf_agents/environments/suite_pybullet_test.py	25
tf_agents/environments/test_envs.py	107
tf_agents/environments/test_envs_test.py	71
tf_agents/environments/tf_environment_test.py	242
tf_agents/environments/tf_py_environment_test.py	348
tf_agents/environments/tf_wrappers_test.py	83
tf_agents/environments/trajectory_replay_test.py	64
tf_agents/environments/utils_test.py	86
tf_agents/environments/wrappers_test.py	1132
tf_agents/eval/metric_utils_test.py	30
tf_agents/experimental/distributed/reverb_variable_container_test.py	201
tf_agents/keras_layers/bias_layer_test.py	29
tf_agents/keras_layers/dynamic_unroll_layer_test.py	124
tf_agents/keras_layers/inner_reshape_test.py	33
tf_agents/keras_layers/permanent_variable_rate_dropout_test.py	34
tf_agents/keras_layers/rnn_wrapper_test.py	60
tf_agents/keras_layers/squashed_outer_wrapper_test.py	34
tf_agents/metrics/batched_py_metric_test.py	84
tf_agents/metrics/metric_equality_test.py	95
tf_agents/metrics/py_metric_test.py	163
tf_agents/metrics/py_metrics_test.py	213
tf_agents/metrics/tf_metrics_test.py	343
tf_agents/metrics/tf_py_metric_test.py	86
tf_agents/networks/actor_distribution_network_test.py	128
tf_agents/networks/actor_distribution_rnn_network_test.py	168
tf_agents/networks/categorical_projection_network_test.py	33
tf_agents/networks/categorical_q_network_test.py	111
tf_agents/networks/encoding_network_test.py	337
tf_agents/networks/mask_splitter_network_test.py	154
tf_agents/networks/nest_map_test.py	161
tf_agents/networks/network_test.py	318
tf_agents/networks/normal_projection_network_test.py	73
tf_agents/networks/q_network_test.py	205
tf_agents/networks/q_rnn_network_test.py	117
tf_agents/networks/sequential_test.py	206
tf_agents/networks/test_utils.py	12
tf_agents/networks/utils_test.py	109
tf_agents/networks/value_network_test.py	56
tf_agents/networks/value_rnn_network_test.py	92
tf_agents/policies/actor_policy_test.py	209
tf_agents/policies/async_policy_saver_test.py	113
tf_agents/policies/batched_py_policy_test.py	88
tf_agents/policies/boltzmann_policy_test.py	76
tf_agents/policies/categorical_q_policy_test.py	178
tf_agents/policies/epsilon_greedy_policy_test.py	170
tf_agents/policies/fixed_policy_test.py	73
tf_agents/policies/gaussian_policy_test.py	106
tf_agents/policies/greedy_policy_test.py	73
tf_agents/policies/ou_noise_policy_test.py	106
tf_agents/policies/policy_info_updater_wrapper_test.py	78
tf_agents/policies/policy_loader_test.py	63
tf_agents/policies/policy_saver_test.py	835
tf_agents/policies/py_epsilon_greedy_policy_test.py	111
tf_agents/policies/py_tf_eager_policy_test.py	254
tf_agents/policies/py_tf_policy_test.py	263
tf_agents/policies/q_policy_test.py	154
tf_agents/policies/qtopt_cem_policy_test.py	318
tf_agents/policies/random_py_policy_test.py	93
tf_agents/policies/random_tf_policy_test.py	275
tf_agents/policies/samplers/cem_actions_sampler_continuous_and_one_hot_test.py	128
tf_agents/policies/samplers/cem_actions_sampler_continuous_test.py	53
tf_agents/policies/samplers/cem_actions_sampler_hybrid_test.py	43
tf_agents/policies/scripted_py_policy_test.py	155
tf_agents/policies/temporal_action_smoothing_test.py	68
tf_agents/policies/tf_policy_test.py	244
tf_agents/policies/tf_py_policy_test.py	134
tf_agents/replay_buffers/episodic_replay_buffer_driver_test.py	89
tf_agents/replay_buffers/episodic_replay_buffer_test.py	1174
tf_agents/replay_buffers/episodic_table_test.py	248
tf_agents/replay_buffers/py_replay_buffers_test.py	201
tf_agents/replay_buffers/replay_buffer_test.py	44
tf_agents/replay_buffers/reverb_replay_buffer_test.py	414
tf_agents/replay_buffers/reverb_utils_test.py	470
tf_agents/replay_buffers/table_test.py	245
tf_agents/replay_buffers/tf_uniform_replay_buffer_test.py	499
tf_agents/specs/__init__.py	14
tf_agents/specs/array_spec.py	215
tf_agents/specs/array_spec_test.py	434
tf_agents/specs/bandit_spec_utils.py	48
tf_agents/specs/distribution_spec.py	44
tf_agents/specs/distribution_spec_test.py	26
tf_agents/specs/specs_test.py	39
tf_agents/specs/tensor_spec.py	308
tf_agents/specs/tensor_spec_test.py	381
tf_agents/system/multiprocessing_test.py	56
tf_agents/tf_agents_api_test.py	16
tf_agents/train/actor_test.py	118
tf_agents/train/learner_test.py	363
tf_agents/train/ppo_learner_test.py	231
tf_agents/train/tpu_ppo_learner_test.py	64
tf_agents/train/utils/replay_buffer_utils_test.py	48
tf_agents/train/utils/spec_utils_test.py	23
tf_agents/train/utils/strategy_utils_test.py	33
tf_agents/train/utils/test_utils.py	249
tf_agents/train/utils/train_utils_test.py	101
tf_agents/trajectories/policy_step_test.py	32
tf_agents/trajectories/test_utils.py	12
tf_agents/trajectories/time_step_test.py	342
tf_agents/trajectories/trajectory_test.py	276
tf_agents/utils/common_members_not_overridden_test.py	68
tf_agents/utils/common_test.py	839
tf_agents/utils/composite_test.py	42
tf_agents/utils/eager_utils_test.py	486
tf_agents/utils/example_encoding_dataset_test.py	137
tf_agents/utils/example_encoding_test.py	112
tf_agents/utils/nest_utils_test.py	674
tf_agents/utils/numpy_storage_test.py	27
tf_agents/utils/object_identity_test.py	67
tf_agents/utils/session_utils_test.py	71
tf_agents/utils/tensor_normalizer_test.py	343
tf_agents/utils/test_utils.py	34
tf_agents/utils/test_utils_test.py	24
tf_agents/utils/value_ops_test.py	154
tf_agents/utils/xla_test.py	41
tools/graph_builder_test.py	35
tools/test_colabs.py	96