File tree Expand file tree Collapse file tree 3 files changed +3
-3
lines changed
com.unity.ml-agents/Runtime Expand file tree Collapse file tree 3 files changed +3
-3
lines changed Original file line number Diff line number Diff 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
Original file line number Diff line number Diff 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
Original file line number Diff line number Diff 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 ;
You can’t perform that action at this time.
0 commit comments