Update keras-metadata.json

source/keras.js

@@ -1140,15 +1140,8 @@ keras.Metadata = class {
         this._map = new Map();
         this._attributeCache = 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 ]));
         }
     }
 

tools/keras-script.py

@@ -140,9 +140,8 @@ def metadata():
     json_root = json.loads(json_file.read())
     json_file.close()
 
-    for entry in json_root:
-        name = entry['name']
-        schema = entry['schema']
+    for schema in json_root:
+        name = schema['name']
         if 'package' in schema:
             class_name = schema['package'] + '.' + name
             class_definition = pydoc.locate(class_name)
@@ -182,7 +181,7 @@ def metadata():
                     raise Exception('')
 
     json_file = open(json_path, 'w')
-    json_data = json.dumps(json_root, sort_keys=True, indent=2)
+    json_data = json.dumps(json_root, sort_keys=False, indent=2)
     for line in json_data.splitlines():
         json_file.write(line.rstrip() + '\n')
     json_file.close()