Update tensorflow

PiperOrigin-RevId: 503968820
Change-Id: I2ab071d6872bc2f9eaab4a464aed684036b966d2

requirements.txt

@@ -9,6 +9,5 @@ jax==0.3.25
 ml-collections==0.1.0
 numpy==1.21.6
 pandas==1.3.4
-protobuf==3.20.1
 scipy==1.7.0
-tensorflow-cpu==2.9.0
+tensorflow-cpu==2.11.0

run_alphafold.py

@@ -161,7 +161,7 @@ def _check_flag(flag_name: str,
                      f'"--{other_flag_name}={FLAGS[other_flag_name].value}".')
 
 
-def _jnp_to_np(output: dict[str, Any]) -> dict[str, Any]:
+def _jnp_to_np(output: Dict[str, Any]) -> Dict[str, Any]:
   """Recursively changes jax arrays to numpy arrays."""
   for k, v in output.items():
     if isinstance(v, dict):