stable-baselines3/tests
Alex Pasquali b702884c23
Removed shared layers in mlp_extractor (#1292)
* Modified actor-critic policies & MlpExtractor class

ActorCriticPolicy:
  - changed type hint of net_arch param: now it's a dict
  - removed check that if features extractor is not shared: no shared layers are allowed in the mlp_extractor regardless of the features extractor
ActorCriticCnnPolicy:
  - changed type hint of net_arch param: now it's a dict
MultiInputActorcriticPolicy:
  - changed type hint of net_arch param: now it's a dict
MlpExtractor:
  - changed type hint of net_arch param: now it's a dict
  - adapted networks creation
  - adapted methods: forward, forward_actor & forward_critic

* Removed shared layers in mlp_extractor

* Updated docs and changelog + reformat

* Updated custom policy tests

* Removed test on deprecation warning for share layers in mlp_extractor

Now shared layers are removed

* Update version

* Update RL Zoo doc

* Fix linter warnings

* Add ruff to Makefile (experimental)

* Add backward compat code and minor updates

* Update tests

* Add backward compatibility

* Fix test

* Improve compat code

Co-authored-by: Antonin RAFFIN <antonin.raffin@ensta.org>
2023-01-23 14:55:19 +01:00
..
__init__.py Init: TD3 2019-09-05 17:29:41 +02:00
test_buffers.py Support for device=auto buffers and set it as default value (#1009) 2022-08-16 17:54:55 +02:00
test_callbacks.py env_id consistency in tests (#1224) 2022-12-20 16:01:26 +01:00
test_cnn.py Modified ActorCriticPolicy to support non-shared features extractor (#1148) 2022-12-20 15:12:05 +01:00
test_custom_policy.py Removed shared layers in mlp_extractor (#1292) 2023-01-23 14:55:19 +01:00
test_deterministic.py Gym fixes - Follow up from #705 (#734) 2022-02-04 15:13:57 -08:00
test_dict_env.py Fix support of image like normalized inputs (#1214) 2022-12-20 13:18:28 +01:00
test_distributions.py Fix tests/test_distributions.py type hint (#1186) 2022-11-29 11:27:59 +01:00
test_env_checker.py Standardize the use of from gym import spaces (#1240) 2023-01-02 14:51:11 +01:00
test_envs.py Drop gym.GoalEnv and other minor changes initally from #780 (#1184) 2022-11-28 18:22:31 +01:00
test_gae.py Standardize the use of from gym import spaces (#1240) 2023-01-02 14:51:11 +01:00
test_her.py Upgrade code to Python 3.7+ syntax using pyupgrade (#887) 2022-04-25 13:01:38 +03:00
test_identity.py Removed shared layers in mlp_extractor (#1292) 2023-01-23 14:55:19 +01:00
test_logger.py Standardize the use of from gym import spaces (#1240) 2023-01-02 14:51:11 +01:00
test_monitor.py Automatically create missing directories of `filenames passed to ResultsWriter` (#1072) 2022-09-21 13:14:38 +02:00
test_predict.py Standardize the use of from gym import spaces (#1240) 2023-01-02 14:51:11 +01:00
test_preprocessing.py Add support for multidimensional spaces.MultiBinary observations (#1179) 2022-12-08 18:46:41 +01:00
test_run.py Remove deprecated features and attributes (#1104) 2022-10-11 10:55:16 +02:00
test_save_load.py Hotfix to load policies saved with SB3 <= v1.6 (#1234) 2022-12-22 23:58:30 +01:00
test_sde.py Remove deprecated features and attributes (#1104) 2022-10-11 10:55:16 +02:00
test_spaces.py Standardize the use of from gym import spaces (#1240) 2023-01-02 14:51:11 +01:00
test_tensorboard.py Fix test_vec_normalize.py, test_tensorboard.py and common/monitor.py type hint (#1194) 2023-01-13 18:28:22 +01:00
test_train_eval_mode.py Remove deprecated features and attributes (#1104) 2022-10-11 10:55:16 +02:00
test_utils.py Drop gym.GoalEnv and other minor changes initally from #780 (#1184) 2022-11-28 18:22:31 +01:00
test_vec_check_nan.py Upgrade code to Python 3.7+ syntax using pyupgrade (#887) 2022-04-25 13:01:38 +03:00
test_vec_envs.py Standardize the use of from gym import spaces (#1240) 2023-01-02 14:51:11 +01:00
test_vec_extract_dict_obs.py Support for VecMonitor for gym3-style environments (#311) 2021-04-13 18:09:31 +02:00
test_vec_monitor.py Drop gym.GoalEnv and other minor changes initally from #780 (#1184) 2022-11-28 18:22:31 +01:00
test_vec_normalize.py Fix test_vec_normalize.py, test_tensorboard.py and common/monitor.py type hint (#1194) 2023-01-13 18:28:22 +01:00