Hugging Face's logo

AnnaMats
/
ppo-Pyramids-Training

Reinforcement Learning
ml-agents
TensorBoard
ONNX
Pyramids
deep-reinforcement-learning
ML-Agents-Pyramids
Model card Files Metrics Community
ppo-Pyramids-Training
File size: 2,643 Bytes 
05c9ac2
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
 
using NUnit.Framework;
using Unity.MLAgents.Sensors;


namespace Unity.MLAgents.Tests
{
    [TestFixture]
    public class ObservationSpecTests
    {
        [Test]
        public void TestVectorObsSpec()
        {
            var obsSpec = ObservationSpec.Vector(5);
            Assert.AreEqual(1, obsSpec.Rank);

            var shape = obsSpec.Shape;
            Assert.AreEqual(1, shape.Length);
            Assert.AreEqual(5, shape[0]);

            var dimensionProps = obsSpec.DimensionProperties;
            Assert.AreEqual(1, dimensionProps.Length);
            Assert.AreEqual(DimensionProperty.None, dimensionProps[0]);

            Assert.AreEqual(ObservationType.Default, obsSpec.ObservationType);
        }

        [Test]
        public void TestVariableLengthObsSpec()
        {
            var obsSpec = ObservationSpec.VariableLength(5, 6);
            Assert.AreEqual(2, obsSpec.Rank);

            var shape = obsSpec.Shape;
            Assert.AreEqual(2, shape.Length);
            Assert.AreEqual(5, shape[0]);
            Assert.AreEqual(6, shape[1]);

            var dimensionProps = obsSpec.DimensionProperties;
            Assert.AreEqual(2, dimensionProps.Length);
            Assert.AreEqual(DimensionProperty.VariableSize, dimensionProps[0]);
            Assert.AreEqual(DimensionProperty.None, dimensionProps[1]);

            Assert.AreEqual(ObservationType.Default, obsSpec.ObservationType);
        }

        [Test]
        public void TestVisualObsSpec()
        {
            var obsSpec = ObservationSpec.Visual(5, 6, 7);
            Assert.AreEqual(3, obsSpec.Rank);

            var shape = obsSpec.Shape;
            Assert.AreEqual(3, shape.Length);
            Assert.AreEqual(5, shape[0]);
            Assert.AreEqual(6, shape[1]);
            Assert.AreEqual(7, shape[2]);

            var dimensionProps = obsSpec.DimensionProperties;
            Assert.AreEqual(3, dimensionProps.Length);
            Assert.AreEqual(DimensionProperty.TranslationalEquivariance, dimensionProps[0]);
            Assert.AreEqual(DimensionProperty.TranslationalEquivariance, dimensionProps[1]);
            Assert.AreEqual(DimensionProperty.None, dimensionProps[2]);

            Assert.AreEqual(ObservationType.Default, obsSpec.ObservationType);
        }

        [Test]
        public void TestMismatchShapeDimensionPropThrows()
        {
            var shape = new InplaceArray<int>(1, 2);
            var dimProps = new InplaceArray<DimensionProperty>(DimensionProperty.TranslationalEquivariance);
            Assert.Throws<UnityAgentsException>(() => new ObservationSpec(shape, dimProps));
        }
    }
}