Reformat requirement_mappings_multiversion.json and add tensorflow-pr…

…obability dependency to tensorflow [skip ci]
dhanush-2501 · Sep 13, 2023 · dfcbb15 · dfcbb15
1 parent 81cc410
commit dfcbb15
Showing 1 changed file with 32 additions and 9 deletions.
diff --git a/docker/requirement_mappings_multiversion.json b/docker/requirement_mappings_multiversion.json
@@ -1,11 +1,34 @@
-
 {
-    "tensorflow": [
-        {"tensorflow-probability":{"2.12.0":"0.20.0","2.11.0":"0.19.0"}}
-    ],
-    "jax": ["dm-haiku", "flax",{"jaxlib": {"0.4.10": "0.4.10","0.4.8": "0.4.7"}}],
-    "numpy": ["numpy"],
-    "paddle": ["paddlepaddle"],
-    "mxnet": ["mxnet"],
-    "torch": ["torch-scatter"]
+  "tensorflow": [
+    {
+      "tensorflow-probability": {
+        "2.13.0": "0.21.0",
+        "2.12.0": "0.20.0",
+        "2.11.0": "0.19.0"
+      }
+    }
+  ],
+  "jax": [
+    "dm-haiku",
+    "flax",
+    {
+      "jaxlib": {
+        "0.4.14": "0.4.14",
+        "0.4.10": "0.4.10",
+        "0.4.8": "0.4.7"
+      }
+    }
+  ],
+  "numpy": [
+    "numpy"
+  ],
+  "paddle": [
+    "paddlepaddle"
+  ],
+  "mxnet": [
+    "mxnet"
+  ],
+  "torch": [
+    "torch-scatter"
+  ]
 }