Merge pull request #30 from adrhill:ah/drop_py_constraint

PiperOrigin-RevId: 447717861
Change-Id: If7b2da158d56a4dced6d93122b8b00d4738db9f5

setup.py

@@ -41,8 +41,8 @@ def _get_version():
     keywords='jax graph neural networks python machine learning',
     packages=find_namespace_packages(exclude=['*_test.py']),
     package_data={'jraph': ['ogb_examples/test_data/*']},
+    python_requires='>=3.6',
     install_requires=[
-        'python_version>="3.6"',
         'jax>=0.1.55',
         'jaxlib>=0.1.37',
         'numpy>=1.18.0',
@@ -53,7 +53,6 @@ def _get_version():
                     'ogb_examples': ['dm-haiku>=0.0.2', 'absl-py>=0.9',
                                      'optax>=0.0.1', 'pandas>=1.0.5',
                                      'dm-tree>=0.1.5']},
-    python_requires='>=3.6',
     classifiers=[
         'Development Status :: 5 - Production/Stable',
         'Environment :: Console',