Skip to content

Commit eb33b95

Browse files
authored
Rename policy class (#308)
1 parent 92a4fc1 commit eb33b95

File tree

5 files changed

+6
-6
lines changed

5 files changed

+6
-6
lines changed
Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -5,7 +5,7 @@ function sampleNormal(mean, stdDev) {
55
return mean + z * stdDev;
66
}
77

8-
class NeuralFramePolicy {
8+
class MLPPolicy {
99
constructor(args = {}) {
1010
if (args.system == null) {
1111
throw new Error("system required to create policy");
@@ -128,4 +128,4 @@ class NeuralFramePolicy {
128128
}
129129
}
130130

131-
module.exports = NeuralFramePolicy;
131+
module.exports = MLPPolicy;

algovivo/nn/index.js

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,3 +1,3 @@
11
module.exports = {
2-
NeuralFramePolicy: require("./NeuralFramePolicy")
2+
MLPPolicy: require("./MLPPolicy")
33
}

demo/src/AgentSystem.js

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -23,7 +23,7 @@ export default class AgentSystem {
2323

2424
this.system.set(mesh);
2525
if (policy != null) {
26-
this.policy = new this.algovivo.nn.NeuralFramePolicy({ system: this.system });
26+
this.policy = new this.algovivo.nn.MLPPolicy({ system: this.system });
2727
this.policy.loadData(policy);
2828
}
2929
}

test/nn/generateTrajectory.js

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -31,7 +31,7 @@ async function main() {
3131
triangles: meshData.triangles,
3232
trianglesRsi: meshData.rsi
3333
});
34-
const policy = new algovivo.nn.NeuralFramePolicy({
34+
const policy = new algovivo.nn.MLPPolicy({
3535
system: system,
3636
active: true
3737
});

test/nn/trajectory.test.js

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -34,7 +34,7 @@ test("neural frame policy", async () => {
3434
});
3535
expect(system.l0.toArray()).toEqual(meshData.l0);
3636
expect(system.rsi.toArray()).toEqual(meshData.rsi);
37-
const policy = new algovivo.nn.NeuralFramePolicy({
37+
const policy = new algovivo.nn.MLPPolicy({
3838
system: system,
3939
stochastic: false,
4040
active: true

0 commit comments

Comments
 (0)