Update vscode configs to use pyink formatter

PiperOrigin-RevId: 501585033

Author: Conchylicultor

.style.yapf

@@ -0,0 +1,5 @@
+{
+  "recommendations": [
+      "ms-python.black-formatter"
+  ]
+}

.vscode/extensions.json

@@ -3,21 +3,21 @@
     "files.trimFinalNewlines": true,
     "files.trimTrailingWhitespace": true,
     "files.associations": {
-        ".pylintrc": "ini",
-        ".style.yapf": "ini"
+        ".pylintrc": "ini"
     },
     "python.testing.unittestEnabled": false,
     "python.testing.nosetestsEnabled": false,
     "python.testing.pytestEnabled": true,
     "python.linting.pylintUseMinimalCheckers": false,
     "[python]": {
-        "editor.rulers": [
-            80
-        ],
+        "editor.rulers": [80],
         "editor.tabSize": 2,
+        "editor.defaultFormatter": "ms-python.black-formatter",
+        "editor.formatOnSave": true,
         "editor.detectIndentation": false
     },
-    "python.formatting.provider": "yapf",
+    "python.formatting.provider": "none",
+    "black-formatter.path": ["pyink"],
     "files.watcherExclude": {
         "**/.git/**": true
     },

.vscode/settings.json

@@ -38,13 +38,18 @@ dev = [
     "pytest",
     "pytest-xdist",
     "pylint>=2.6.0",
-    "yapf",
+    "pyink",
     # Lazy deps
     "chex",
     "jax[cpu]",
     "tf-nightly",
 ]
 
+[tool.pyink]
+# Formatting configuration to follow Google style-guide
+pyink-indentation = 2
+pyink-use-majority-quotes = true
+
 [build-system]
 requires = ["flit_core >=3.5,<4"]
 build-backend = "flit_core.buildapi"