Skip to content

Commit 7d1ba73

Browse files
committed
Actually hook up the reading and writing of capabilities in C#.
1 parent 0311ab2 commit 7d1ba73

File tree

3 files changed

+3
-3
lines changed

3 files changed

+3
-3
lines changed

com.unity.ml-agents/Runtime/Academy.cs

Lines changed: 0 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -131,8 +131,6 @@ public bool IsCommunicatorOn
131131
// Random seed used for inference.
132132
int m_InferenceSeed;
133133

134-
// The Capabilities of the RL trainer.
135-
136134
/// <summary>
137135
/// Set the random seed used for inference. This should be set before any Agents are added
138136
/// to the scene. The seed is passed to the ModelRunner constructor, and incremented each

com.unity.ml-agents/Runtime/Communicator/GrpcExtensions.cs

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -166,6 +166,7 @@ public static UnityRLInitParameters ToUnityRLInitParameters(this UnityRLInitiali
166166
seed = inputProto.Seed,
167167
pythonLibraryVersion = inputProto.PackageVersion,
168168
pythonCommunicationVersion = inputProto.CommunicationVersion,
169+
TrainerCapabilities = inputProto.Capabilities.ToRLCapabilities()
169170
};
170171
}
171172

com.unity.ml-agents/Runtime/Communicator/RpcCommunicator.cs

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -108,7 +108,8 @@ public UnityRLInitParameters Initialize(CommunicatorInitParameters initParameter
108108
{
109109
Name = initParameters.name,
110110
PackageVersion = initParameters.unityPackageVersion,
111-
CommunicationVersion = initParameters.unityCommunicationVersion
111+
CommunicationVersion = initParameters.unityCommunicationVersion,
112+
Capabilities = initParameters.CSharpCapabilities.ToProto()
112113
};
113114

114115
UnityInputProto input;

0 commit comments

Comments
 (0)