Fix install dependencies

Author: master

requirements.txt

@@ -1 +1,2 @@
 tensorflow==2.6.3
+protobuf==3.19.6

setup.py

@@ -3,14 +3,14 @@
 
 setup(
     name="tensorflow-riemopt",
-    version="0.1.0",
+    version="0.1.2",
     description="a library for optimization on Riemannian manifolds",
     long_description=open("README.md", "r").read(),
     long_description_content_type="text/markdown",
     author="Oleg Smirnov",
     author_email="[email protected]",
     packages=find_packages(),
-    install_requires=Path("requirements.txt").read_text().splitlines(),
+    install_requires=["tensorflow==2.6.3", "protobuf==3.19.6"],
     python_requires=">=3.6.0",
     url="https://github.com/master/tensorflow-riemopt",
     zip_safe=True,

tensorflow_riemopt/manifolds/product.py

@@ -82,7 +82,7 @@ def _get_slice(self, x, idx):
 
     def _product_fn(self, fn, *args, **kwargs):
         results = []
-        for (i, m) in enumerate(self._manifolds):
+        for i, m in enumerate(self._manifolds):
             arg_slices = [self._get_slice(arg, i) for arg in args]
             result = getattr(m, fn)(*arg_slices, **kwargs)
             shape = tf.concat([tf.shape(result)[:-1], [-1]], axis=-1)
@@ -94,7 +94,7 @@ def random(self, shape, dtype=tf.float32):
             raise ValueError("Invalid shape {}".format(shape))
         shape = list(shape)
         results = []
-        for (i, m) in enumerate(self._manifolds):
+        for i, m in enumerate(self._manifolds):
             result = m.random(shape[:-1] + self._shapes[i], dtype=dtype)
             results.append(tf.reshape(result, shape[:-1] + [-1]))
         return tf.concat(results, axis=-1)