Browse Source

Update lasagne-metadata.json

Lutz Roeder 5 years ago
parent
commit
c00699682e
2 changed files with 6 additions and 19 deletions
  1. 3 9
      source/lasagne-metadata.json
  2. 3 10
      source/lasagne.js

+ 3 - 9
source/lasagne-metadata.json

@@ -1,20 +1,14 @@
 [
   {
     "name": "lasagne.layers.conv.Conv2DLayer",
-    "schema": {
-      "category": "Layer"
-    }
+    "category": "Layer"
   },
   {
     "name": "lasagne.layers.pool.MaxPool2DLayer",
-    "schema": {
-      "category": "Pool"
-    }
+    "category": "Pool"
   },
   {
     "name": "lasagne.layers.dense.DenseLayer",
-    "schema": {
-      "category": "Layer"
-    }
+    "category": "Layer"
   }
 ]

+ 3 - 10
source/lasagne.js

@@ -253,20 +253,13 @@ lasagne.Metadata = class {
     constructor(data) {
         this._map = new Map();
         if (data) {
-            const items = JSON.parse(data);
-            if (items) {
-                for (const item of items) {
-                    if (item.name && item.schema) {
-                        item.schema.name = item.name;
-                        this._map.set(item.name, item.schema);
-                    }
-                }
-            }
+            const metadata = JSON.parse(data);
+            this._map = new Map(metadata.map((item) => [ item.name, item ]));
         }
     }
 
     type(name) {
-        return this._map.has(name) ? this._map.get(name) : null;
+        return this._map.get(name);
     }
 
     attribute(type, name) {