Fix code style

Author: master

examples/tutorial.ipynb

@@ -13,6 +13,7 @@
     "%matplotlib inline\n",
     "\n",
     "import sys\n",
+    "\n",
     "sys.path.append(\"../\")\n",
     "import tensorflow_riemopt as riemopt"
    ]
@@ -119,7 +120,7 @@
     "\n",
     "opt = riemopt.optimizers.RiemannianAdam(learning_rate=0.2)\n",
     "\n",
-    "npole = tf.constant([0., 1.])\n",
+    "npole = tf.constant([0.0, 1.0])\n",
     "phi = np.linspace(-np.pi, np.pi, 100)\n",
     "\n",
     "for _ in range(STEPS):\n",
@@ -136,8 +137,14 @@
     "    plt.plot(np.cos(phi), np.sin(phi))\n",
     "    plt.plot(var_np[:, 0], var_np[:, 1], '+', color='black')\n",
     "    for i in range(len(egrad_np)):\n",
-    "        plt.arrow(var_np[i][0], var_np[i][1], -egrad_np[i][0], -egrad_np[i][1],\n",
-    "                  width=0.01, color='green')\n",
+    "        plt.arrow(\n",
+    "            var_np[i][0],\n",
+    "            var_np[i][1],\n",
+    "            -egrad_np[i][0],\n",
+    "            -egrad_np[i][1],\n",
+    "            width=0.01,\n",
+    "            color='green',\n",
+    "        )\n",
     "    plt.plot(var_t_np[:, 0], var_t_np[:, 1], '+', color='red')\n",
     "    plt.show()"
    ]
@@ -152,7 +159,7 @@
  ],
  "metadata": {
   "kernelspec": {
-   "display_name": "Python 3",
+   "display_name": "Python 3 (ipykernel)",
    "language": "python",
    "name": "python3"
   },
@@ -166,7 +173,7 @@
    "name": "python",
    "nbconvert_exporter": "python",
    "pygments_lexer": "ipython3",
-   "version": "3.7.3"
+   "version": "3.10.6"
   }
  },
  "nbformat": 4,

examples/usage.ipynb

@@ -14,6 +14,7 @@
     "from mpl_toolkits.mplot3d import Axes3D\n",
     "\n",
     "import sys\n",
+    "\n",
     "sys.path.append(\"../\")\n",
     "import tensorflow_riemopt as riemopt"
    ]
@@ -32,15 +33,17 @@
     "    s_z = np.outer(np.abs(np.cos(phi)), np.ones_like(psi))\n",
     "    return ax.plot_wireframe(s_x, s_y, s_z, color=color, alpha=0.3)\n",
     "\n",
+    "\n",
     "def plot_vector(ax, x, u, color=\"darkorange\"):\n",
     "    return ax.quiver(*x, *u, length=0.6, normalize=True, color=color)\n",
     "\n",
+    "\n",
     "def plot_hyperplane(ax, p, u, v, color=\"limegreen\"):\n",
-    "    xx = np.linspace(-0.05, 1., 10)\n",
-    "    yy = np.linspace(-1., 0.1, 10)\n",
+    "    xx = np.linspace(-0.05, 1.0, 10)\n",
+    "    yy = np.linspace(-1.0, 0.1, 10)\n",
     "    x, y = np.meshgrid(xx, yy)\n",
     "    n = np.cross(u, v)\n",
-    "    z = (- n[0] * x - n[1] * y + p.dot(n)) / n[2]\n",
+    "    z = (-n[0] * x - n[1] * y + p.dot(n)) / n[2]\n",
     "    return ax.plot_wireframe(x, y, z, color=color, alpha=0.4)"
    ]
   },
@@ -53,7 +56,7 @@
     "S = riemopt.manifolds.Sphere()\n",
     "\n",
     "x = S.projx(tf.constant([0.1, -0.1, 0.1]))\n",
-    "u = S.proju(x, tf.constant([1., 1., 1.]))\n",
+    "u = S.proju(x, tf.constant([1.0, 1.0, 1.0]))\n",
     "v = S.proju(x, tf.constant([-0.7, -1.4, 1.4]))\n",
     "y = S.exp(x, v)\n",
     "u_ = S.transp(x, y, u)\n",
@@ -866,9 +869,9 @@
     "fig = plt.figure()\n",
     "ax = fig.gca(projection=\"3d\")\n",
     "ax.axis(\"off\")\n",
-    "ax.set_zlim3d(-1.5, 1.5) \n",
-    "ax.set_ylim3d(-1.5, 1.5) \n",
-    "ax.set_xlim3d(-1.5, 1.5) \n",
+    "ax.set_zlim3d(-1.5, 1.5)\n",
+    "ax.set_ylim3d(-1.5, 1.5)\n",
+    "ax.set_xlim3d(-1.5, 1.5)\n",
     "\n",
     "plot_halfsphere(ax)\n",
     "plot_hyperplane(ax, x, u, v)\n",
@@ -885,7 +888,7 @@
  ],
  "metadata": {
   "kernelspec": {
-   "display_name": "Python 3",
+   "display_name": "Python 3 (ipykernel)",
    "language": "python",
    "name": "python3"
   },
@@ -899,7 +902,7 @@
    "name": "python",
    "nbconvert_exporter": "python",
    "pygments_lexer": "ipython3",
-   "version": "3.7.3"
+   "version": "3.10.6"
   }
  },
  "nbformat": 4,

tensorflow_riemopt/optimizers/constrained_rmsprop.py

@@ -87,9 +87,7 @@ def _create_slots(self, var_list):
                 self.add_slot(var, "mg")
 
     def _prepare_local(self, var_device, var_dtype, apply_state):
-        super()._prepare_local(
-            var_device, var_dtype, apply_state
-        )
+        super()._prepare_local(var_device, var_dtype, apply_state)
 
         rho = array_ops.identity(self._get_hyper("rho", var_dtype))
         apply_state[(var_device, var_dtype)].update(

tensorflow_riemopt/optimizers/riemannian_adam.py

@@ -92,9 +92,7 @@ def _create_slots(self, var_list):
                 self.add_slot(var, "vhat")
 
     def _prepare_local(self, var_device, var_dtype, apply_state):
-        super()._prepare_local(
-            var_device, var_dtype, apply_state
-        )
+        super()._prepare_local(var_device, var_dtype, apply_state)
 
         local_step = math_ops.cast(self.iterations + 1, var_dtype)
         beta_1_t = array_ops.identity(self._get_hyper("beta_1", var_dtype))

tensorflow_riemopt/optimizers/riemannian_gradient_descent.py

@@ -85,9 +85,7 @@ def _create_slots(self, var_list):
                 self.add_slot(var, "momentum")
 
     def _prepare_local(self, var_device, var_dtype, apply_state):
-        super()._prepare_local(
-            var_device, var_dtype, apply_state
-        )
+        super()._prepare_local(var_device, var_dtype, apply_state)
         apply_state[(var_device, var_dtype)]["momentum"] = array_ops.identity(
             self._get_hyper("momentum", var_dtype)
         )