Merge branch 'develop' into seb-ff-ippo-only

instadeepai · Nov 13, 2024 · b6712e2 · b6712e2
2 parents 3fce221 + 6092dc6
commit b6712e2
Show file tree

Hide file tree

Showing 3 changed files with 1 addition and 2 deletions.
diff --git a/mava/utils/logger.py b/mava/utils/logger.py
@@ -345,7 +345,6 @@ def get_logger_path(config: DictConfig, logger_type: str) -> str:
 
 def describe(x: ArrayLike) -> Union[Dict[str, ArrayLike], ArrayLike]:
     """Generate summary statistics for an array of metrics (mean, std, min, max)."""
-
     if not isinstance(x, (jax.Array, np.ndarray)) or x.ndim == 0:
         return x
 

diff --git a/mava/utils/make_env.py b/mava/utils/make_env.py
@@ -57,6 +57,7 @@
     UoeWrapper,
     VectorConnectorWrapper,
     async_multiagent_worker,
+    VectorConnectorWrapper,
 )
 
 # Registry mapping environment names to their generator and wrapper classes.

diff --git a/mava/utils/network_utils.py b/mava/utils/network_utils.py
@@ -20,7 +20,6 @@
 _DISCRETE = "discrete"
 _CONTINUOUS = "continuous"
 
-
 def get_action_head(action_types: Union[Spec, Space]) -> Tuple[Dict[str, str], str]:
     """Returns the appropriate action head config based on the environment action_spec."""
     if isinstance(action_types, (DiscreteArray, MultiDiscreteArray, Discrete, MultiDiscrete)):