Update continuous critic type annotation

DLR-RM · Apr 13, 2023 · e613373 · e613373
1 parent eafd84c
commit e613373
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/stable_baselines3/common/policies.py b/stable_baselines3/common/policies.py
@@ -898,10 +898,12 @@ class ContinuousCritic(BaseModel):
         between the actor and the critic (this saves computation time)
     """
 
+    features_extractor: BaseFeaturesExtractor
+
     def __init__(
         self,
         observation_space: spaces.Space,
-        action_space: spaces.Space,
+        action_space: spaces.Box,
         net_arch: List[int],
         features_extractor: nn.Module,
         features_dim: int,