Allow dev and github main versions to depend on corresponding Tensorflow

dev+main versions.

PiperOrigin-RevId: 644111145

Author: vkarampudi

setup.py

@@ -311,7 +311,12 @@ def select_constraint(default, nightly=None, git_master=None):
         'sacrebleu>=2.3,<4',
         'scipy>=1.4.1,<2',
         'six>=1.12,<2',
-        'tensorflow>=2.15,<2.16',
+        'tensorflow'
+        + select_constraint(
+            default='>=2.15,<2.16',
+            nightly='>=2.16.0.dev',
+            git_master='@git+https://github.com/tensorflow/tensorflow@master',
+        ),
         'tensorflow-estimator>=2.10',
         'tensorflow-metadata'
         + select_constraint(