format files.

Author: surfnerd

com.unity.ml-agents/CHANGELOG.md

@@ -80,6 +80,7 @@ and this project adheres to
 - `num_updates` and `train_interval` for SAC were replaced with `steps_per_update`. (#3690)
 - `WriteAdapter` was renamed to `ObservationWriter`. If you have a custom `ISensor` implementation,
 you will need to change the signature of its `Write()` method. (#3834)
+- `UnityRLCapabilities` was added to help inform users when RL features are mismatched between C# and Python packages. (#3831)
 
 ### Bug Fixes
 

ml-agents-envs/mlagents_envs/environment.py

@@ -120,16 +120,18 @@ def get_capabilities_proto() -> UnityRLCapabilitiesProto:
         return capabilities
 
     @staticmethod
-    def check_csharp_capabitlities(unity_package_ver: str, python_package_ver: str, caps: UnityRLCapabilitiesProto):
+    def warn_csharp_base_capabitlities(
+        caps: UnityRLCapabilitiesProto, unity_package_ver: str, python_package_ver: str
+    ) -> None:
         if not caps.BaseRLCapabilities:
-            logger.warning("WARNING: The Unity process is not running with the expected base Reinforcement Learning"
-                           " capabilities. Please be sure upgrade the Unity Package to a version that is compatible with this "
-                           "python package.\n"
-                           f"Python package version: {python_package_ver}, C# package version: {unity_package_ver}"
-                           f"Please find the versions that work best together from our release page.\n"
-                           "https://github.com/Unity-Technologies/ml-agents/releases")
-
-
+            logger.warning(
+                "WARNING: The Unity process is not running with the expected base Reinforcement Learning"
+                " capabilities. Please be sure upgrade the Unity Package to a version that is compatible with this "
+                "python package.\n"
+                f"Python package version: {python_package_ver}, C# package version: {unity_package_ver}"
+                f"Please find the versions that work best together from our release page.\n"
+                "https://github.com/Unity-Technologies/ml-agents/releases"
+            )
 
     def __init__(
         self,
@@ -222,7 +224,11 @@ def __init__(
             self._close(0)
             UnityEnvironment._raise_version_exception(aca_params.communication_version)
 
-        UnityEnvironment.check_csharp_capabilities(aca_params.capabilities)
+        UnityEnvironment.warn_csharp_base_capabitlities(
+            aca_params.capabilities,
+            aca_params.package_version,
+            UnityEnvironment.API_VERSION,
+        )
 
         self._env_state: Dict[str, Tuple[DecisionSteps, TerminalSteps]] = {}
         self._env_specs: Dict[str, BehaviorSpec] = {}