/// <summary>
    ///  Gathering inputs.
    /// </summary>
    /// <returns></returns>
    public override List <float> CollectState()
    {
        // Total 43 inputs
        List <float> state = _ballBehavior.CollectBallState(); // 40 inputs collected

        state.Add(transform.localEulerAngles.x / 360f);
        state.Add(transform.localEulerAngles.z / 360f);
        state.Add(Convert.ToSingle(_ballBehavior.IsCornered));
        return(state);
    }
    /// <summary>
    ///  Gathering inputs.
    /// </summary>
    public override void CollectObservations()
    {
        // Total 43 inputs
        List <float> state = _ballBehavior.CollectBallState(); // 40 inputs collected

        state.Add(transform.localEulerAngles.x / 360f);
        state.Add(transform.localEulerAngles.z / 360f);
        state.Add(Convert.ToSingle(_ballBehavior.IsCornered));
        AddVectorObs(state);
    }