|
using UnityEngine; |
|
using UnityEditor; |
|
|
|
namespace Unity.MLAgents.Editor |
|
{ |
|
|
|
|
|
|
|
[CustomEditor(typeof(Agent), true)] |
|
[CanEditMultipleObjects] |
|
internal class AgentEditor : UnityEditor.Editor |
|
{ |
|
public override void OnInspectorGUI() |
|
{ |
|
var serializedAgent = serializedObject; |
|
serializedAgent.Update(); |
|
|
|
var maxSteps = serializedAgent.FindProperty("MaxStep"); |
|
|
|
EditorGUILayout.PropertyField( |
|
maxSteps, |
|
new GUIContent("Max Step", "The per-agent maximum number of steps.") |
|
); |
|
|
|
serializedAgent.ApplyModifiedProperties(); |
|
|
|
EditorGUILayout.LabelField("", GUI.skin.horizontalSlider); |
|
base.OnInspectorGUI(); |
|
} |
|
} |
|
} |
|
|