diff --git a/src/setup.py b/src/setup.py
index 00684da..1f66c68 100644
--- a/src/setup.py
+++ b/src/setup.py
@@ -11,7 +11,7 @@
     url="https://github.com/SUwonglab/CausalBGM",
     packages=setuptools.find_packages(),
     install_requires=[
-   'tensorflow==2.10.0',
+   'tensorflow',
    'pyyaml',
    'scikit-learn',
    'pandas',