fix m1 tf issue

Author: refraction-ray

CHANGELOG.md

@@ -2,6 +2,10 @@
 
 ## Unreleased
 
+### Fixed
+
+- Fix installation issue with tensorflow requirements on MACOS with M1 chip
+
 ## 0.1.1
 
 ### Added

setup.py

@@ -1,10 +1,17 @@
 import setuptools
+import platform
+
 from tensorcircuit import __version__, __author__
 
 
 with open("README.md", "r") as fh:
     long_description = fh.read()
 
+install_requires = ["numpy", "scipy", "tensornetwork", "networkx"]
+
+if platform.processor() != "arm":
+    install_requires.append("tensorflow")
+    # avoid the embarassing macos M1 chip case, where the package is called tensorflow-macos
 
 setuptools.setup(
     name="tensorcircuit",
@@ -17,16 +24,16 @@
     url="https://github.com/refraction-ray/tensorcircuit",
     packages=setuptools.find_packages(),
     include_package_data=True,
-    install_requires=["numpy", "scipy", "tensornetwork", "networkx", "tensorflow"],
+    install_requires=install_requires,
     tests_require=[
         "pytest",
         "pytest-lazy-fixture",
         "pytest-cov",
         "pytest-benchmark",
         "pytest-xdist",
     ],
-    classifiers=(
+    classifiers=[
         "Programming Language :: Python :: 3",
         "Operating System :: OS Independent",
-    ),
+    ],
 )

tensorcircuit/quantum.py

@@ -1377,11 +1377,10 @@ def ps2coo_core(
         )
         return tf.SparseTensor(indices=indices, values=values, dense_shape=(s, s))  # type: ignore
 
-except NameError:
+except (NameError, ImportError):
     logger.warning(
         "tensorflow is not installed, and sparse Hamiltonian generation utilities are disabled"
     )
-    # TODO(@refraction-ray): backend agnostic sparse matrix generation?
 
 # some quantum quatities below