Update ant/weights_tensorflow_signatures/config.json

ant/weights_tensorflow_signatures/config.json

@@ -6,7 +6,7 @@
     "latent_dim": 32
   },
 
-  "models": [
+   "models": [
     {
       "name": "encoder",
       "framework": null,
@@ -21,26 +21,24 @@
     {
       "name": "dynamics",
       "framework": null,
-      "format": "saved_model",
+      "format": "module",
       "source": {
-        "path": "transition_concat.pb"
+        "object_ref": "transition_concat_saved_model"
       },
       "signature": {
-        "inputs": ["z", "a"],
-        "call_mode": "concat"
+        "inputs": ["z", "a"]
       }
     },
     {
       "name": "reward",
       "framework": null,
-      "format": "saved_model",
+      "format": "jit",
       "source": {
-        "path": "reward_positional.pb"
+        "object_ref": "reward_concat_saved_model"
       },
       "signature": {
-        "inputs": ["z", "a"],
-        "call_mode": "positional"
+        "inputs": ["z", "a"]
       }
     }
   ]
-}
+}