detensorflow

Author: refraction-ray

setup.py

@@ -1,16 +1,10 @@
 import setuptools
 
 from tensorcircuit import __version__, __author__
-from tensorcircuit.utils import is_m1mac
 
 with open("README.md", "r") as fh:
     long_description = fh.read()
 
-install_requires = ["numpy", "scipy", "tensornetwork", "networkx"]
-
-if not is_m1mac():
-    install_requires.append("tensorflow")
-    # avoid the embarassing macos M1 chip case, where the package is called tensorflow-macos
 
 setuptools.setup(
     name="tensorcircuit",
@@ -23,14 +17,13 @@
     url="https://github.com/tencent-quantum-lab/tensorcircuit",
     packages=setuptools.find_packages(),
     include_package_data=True,
-    install_requires=install_requires,
-    tests_require=[
-        "pytest",
-        "pytest-lazy-fixture",
-        "pytest-cov",
-        "pytest-benchmark",
-        "pytest-xdist",
-    ],
+    install_requires=["numpy", "scipy", "tensornetwork", "networkx"],
+    extras_require={
+        "tensorflow": ["tensorflow"],
+        "jax": ["jax", "jaxlib"],
+        "torch": ["torch"],
+        "qiskit": ["qiskit"],
+    },
     classifiers=[
         "Programming Language :: Python :: 3",
         "Operating System :: OS Independent",

tensorcircuit/interfaces/tensorflow.py

@@ -5,8 +5,6 @@
 from typing import Any, Callable, Tuple
 from functools import wraps
 
-import tensorflow as tf
-
 from ..cons import backend
 from ..utils import return_partial
 from .tensortrans import general_args_to_backend
@@ -30,6 +28,8 @@ def fun_tf(*x: Any) -> Any:
 
 
 def tf_dtype(dtype: str) -> Any:
+    import tensorflow as tf
+
     if isinstance(dtype, str):
         return getattr(tf, dtype)
     return dtype
@@ -72,6 +72,7 @@ def f(params):
         while AD is also supported
     :rtype: Callable[..., Any]
     """
+    import tensorflow as tf
 
     if jit is True:
         fun = backend.jit(fun)

tensorcircuit/translation.py

@@ -254,7 +254,7 @@ def qiskit2tc(
                 base_gate = gate_info[0].base_gate
                 ctrl_state = [1] * base_gate.num_qubits
                 idx = idx[: -base_gate.num_qubits] + idx[-base_gate.num_qubits :][::-1]
-                print(idx)
+                # print(idx)
                 tc_circuit.multicontrol(
                     *idx, ctrl=ctrl_state, unitary=base_gate.to_matrix()
                 )