Augustin Zidek
79a0210693
Format msa_pairing
...
PiperOrigin-RevId: 794545379
Change-Id: I40ca93f7e6864a5906a40168250261c9c56c59ff
2025-08-13 06:08:01 -07:00
Augustin Zidek
6544d69fc3
Remove unused dependencies
...
PiperOrigin-RevId: 794083897
Change-Id: Iecdc0b5d1cd43f1702752eb3655edc1b9894f520
2025-08-12 06:30:33 -07:00
Augustin Zidek
b1d8619675
Use pytest for testing instead of setup.py test
...
PiperOrigin-RevId: 789706772
Change-Id: Id4d58250e03bc41ab7d9a90c3d569266cb2db4ad
2025-08-01 04:19:07 -07:00
Augustin Zidek
e9b68483fc
Fix Docker build due to changes in Conda and don't limit make parallelism
...
PiperOrigin-RevId: 783683713
Change-Id: I24599012fabbf9bf5b87597906b1ebb5a70c029c
2025-07-16 03:10:25 -07:00
Augustin Zidek
b522811d9e
Avoid use of deprecated jax.util APIs
...
jax.util was deprecated in JAX v0.6.0, and will be removed in JAX v0.7.0.
PiperOrigin-RevId: 759626266
Change-Id: If3dcb9a8151a99ecab1f8ec670bd99bbb31bd5de
2025-05-16 08:33:10 -07:00
Hana Joo
8050b46921
No public description
...
PiperOrigin-RevId: 752767512
Change-Id: I0c29c07cd27b7da3472871e8a8e8c5c257458d80
2025-04-29 09:32:35 -07:00
DeepMind
020cd6d6cb
Add information about PDBe template cutoff date
...
PiperOrigin-RevId: 737580394
Change-Id: Id6d3c3f518682d9171c67ac59ce78bf80c6bb27f
2025-03-17 05:40:18 -07:00
Dariusz Lasecki
2ca41b1991
Deprecate the AlphaFold 2 notebook
...
PiperOrigin-RevId: 720951827
Change-Id: Ie30be1252f6e6dc4038be6b307395f45e638d5ad
2025-01-29 06:32:24 -08:00
DeepMind
24fbdfc3ce
Update JSON format docs with new template on/off, cutoff date fields.
...
PiperOrigin-RevId: 720186008
Change-Id: I53b84c0b52509f93e85cc8705aeba4888398f5bb
2025-01-27 08:48:48 -08:00
Augustin Zidek
93e024bd55
Do not use jnp.clip arguments which are not supported in the pinned JAX version.
...
PiperOrigin-RevId: 719259712
Change-Id: I7eb76868275c862201cc7eb26fede8c987e5b760
2025-01-24 05:09:56 -08:00
Augustin Zidek
11a991ea66
Bump Python in Colab to 3.11.
...
PiperOrigin-RevId: 718812632
Change-Id: I54b901e7a361a63a2c3f366535dbe2cd3e792b99
2025-01-23 04:40:44 -08:00
Jake VanderPlas
fa0c5266b3
jax.numpy.clip: update use of deprecated arguments.
...
- a is now positional-only
- a_min is now min
- a_max is now max
The old argument names have been deprecated since JAX v0.4.27.
PiperOrigin-RevId: 715343439
Change-Id: I50b086b249360c142f42ed4d8e50d48692c11dfa
2025-01-14 05:42:09 -08:00
Augustin Zidek
6350ddd63b
Update Conda to 24.11.1 to fix Docker build failure and small improvements to the Dockerfile.
...
PiperOrigin-RevId: 707479885
Change-Id: Ic3c71229bcf3a935cae86aee64ee7c02220e1085
2024-12-18 03:26:03 -08:00
Martin Huschenbett
bbd8300d6a
Automated Code Change: Suppress pytype errors
...
PiperOrigin-RevId: 700257299
Change-Id: If6ce01bd60ab0422eca10578db8d58b1a3137b40
2024-11-26 01:41:22 -08:00
Augustin Zidek
d95a92aae1
Update the disclaimer to match AlphaFold 3 + run mdformat.
...
PiperOrigin-RevId: 696460542
Change-Id: I42defc0a687207ee57801b0aeba78fe1e63b03e6
2024-11-14 03:34:33 -08:00
Augustin Zidek
85e0be3c7d
Make apt output log-friendly by adding the --quiet flag.
...
PiperOrigin-RevId: 665259372
Change-Id: I516c7780b8a821065bf19fe3c79dba604f2c917c
2024-08-20 02:46:14 -07:00
Augustin Zidek
a2217f031e
Pin setuptools on a version smaller than 72.0.0.
...
PiperOrigin-RevId: 660325717
Change-Id: I11b5cf0b0b7cc8bfa48de170a1653e013d7734a1
2024-08-07 03:59:31 -07:00
Peter Hawkins
692c01c524
[numpy] Fix users of NumPy APIs that are removed in NumPy 2.0.
...
This change migrates users of APIs removed in NumPy 2.0 to their recommended replacements (https://numpy.org/devdocs/numpy_2_0_migration_guide.html ).
PiperOrigin-RevId: 655904142
Change-Id: Idcf2384de70cb27cb92e3f421dc8e0a9a6466507
2024-07-25 04:17:05 -07:00
Augustin Zidek
93a9a042eb
Inline SCOPData to be compatible with future Biopython versions.
...
PiperOrigin-RevId: 648368396
Change-Id: Icfd8135c5602ca81007dd6074a3258d5ae0187ee
2024-07-01 07:55:19 -07:00
Jake VanderPlas
5b0e1f2c55
Replace deprecated jax.tree_* functions with jax.tree.*
...
The top-level `jax.tree_*` aliases have long been deprecated, and will soon be removed. Alternate APIs are in `jax.tree_util`, with shorter aliases in the `jax.tree` submodule, added in JAX version 0.4.25.
PiperOrigin-RevId: 633878846
Change-Id: I282e177b3c9026b805d6eb5e5d6e6b1287f87e55
2024-05-15 03:32:25 -07:00
Augustin Zidek
f251de6613
Add documentation for AlphaFold Server JSON file format.
...
PiperOrigin-RevId: 631787106
Change-Id: Ib58c166a2945d459c1555dd1920963887b8d7c16
2024-05-08 07:04:54 -07:00
Hamish Tomlinson
4d4a1aea99
Update openmm in AlphaFold colab.
...
PiperOrigin-RevId: 625019041
Change-Id: If6e5b23edeb1f06bd91b4daf1c449d12ed407895
2024-04-15 11:08:04 -07:00
Hamish Tomlinson
dbe2a438eb
Update requirements:
...
- Ubuntu to 20.04
- Cuda to 12.2.2
- Tensorflow to 2.16.1
- Jax to 0.4.26
- Openmm to 8.0.0
PiperOrigin-RevId: 624094170
Change-Id: I49391c3f721e93ac8ccd5a4483cdb6f2ec61cd3b
2024-04-12 01:57:58 -07:00
Hamish Tomlinson
7e7de59cd2
Remove deprecated PRNGKeyArray.
...
PiperOrigin-RevId: 623494831
Change-Id: Ie15c1d517fd7b34662d69d05205151d9c7021da4
2024-04-10 08:01:46 -07:00
Hamish Tomlinson
5208f1b70e
Update jax, haiku and chex versions.
...
PiperOrigin-RevId: 623469847
Change-Id: I64a812e00e099f6503b9aa1e0cfde1133ea8a774
2024-04-10 06:04:14 -07:00
Augustin Zidek
6d95d1982e
Fix how we extract mmCIF resolution (fix ordering issue).
...
Thanks to @ljarosch for reporting and suggesting a fix in https://github.com/google-deepmind/alphafold/issues/915 .
PiperOrigin-RevId: 617817805
Change-Id: I46738b248a054d06fe902626cff161fe521e99f1
2024-03-21 06:03:34 -07:00
Hamish Tomlinson
032e2f2673
Update conda to 24.1.2.
...
PiperOrigin-RevId: 611404763
Change-Id: Icc7d637e1de9fa82da76aa8dccd9b9dce8be6246
2024-02-29 03:01:46 -08:00
Hamish Tomlinson
632ef575c6
Uninstall keras in notebook to avoid the keras initializer fault.
...
PiperOrigin-RevId: 603012242
Change-Id: Icf0120e1baf1278862ea62402a86c9eac5932710
2024-01-31 04:50:56 -08:00
Matthew Johnson
4f447319e4
Loosen overly tight numerical tolerances to allow for one-time random value
...
change from https://github.com/google/jax/discussions/18480
PiperOrigin-RevId: 591243740
Change-Id: I86457bfcaae65771ccb4e39d3e4d929335d71652
2023-12-15 07:13:05 -08:00
Peter Hawkins
1fa9200c7f
[JAX] Replace uses of jax.devices("cpu") with jax.local_devices(backend="cpu").
...
An upcoming change to JAX will include non-local (addressable) CPU devices in jax.devices() when JAX is used multicontroller-style, where there are multiple Python processes.
This change preserves the current behavior by replacing uses of jax.devices("cpu"), which previously only returned local devices, with jax.local_devices("cpu"), which will return local devices both now and in the future.
This change is always be safe (i.e., it should always preserve the previous behavior) but it may sometimes be unnecessary if code is never used in a multicontroller setting.
PiperOrigin-RevId: 582680953
Change-Id: I614739de052185c42932ddb91d1d464751442f65
2023-11-15 08:11:45 -08:00
Augustin Zidek
f715f016d8
Revert a change that is not compatible with JAX 0.3.25.
...
PiperOrigin-RevId: 578486539
Change-Id: Id0e535ebf75916b5179791d093ca714e343ccb69
2023-11-01 05:32:02 -07:00
Augustin Zidek
f78c589304
Update wwPDB paths from ftp.wwpdb.org to files.wwpdb.org.
...
See https://www.wwpdb.org/news/news?year=2022#6303be3b707ecd4f63b3d3d9 for the wwPDB announcement.
PiperOrigin-RevId: 575762003
Change-Id: If25ea2777c33b03f03b27d54046c71753cb33038
2023-10-23 03:14:49 -07:00
DeepMind
bbc133b060
Add assertion to aid type checking.
...
PiperOrigin-RevId: 568220760
Change-Id: Ied9e33692bbd7fb8c02b039f64688cc31f005885
2023-09-25 08:04:59 -07:00
Jake VanderPlas
a8875a9c0d
Replace deprecated PRNGKeyArray isinstance checks
...
For details, see https://github.com/google/jax/pull/17297
PiperOrigin-RevId: 565705637
Change-Id: I88c4a3003166580e28111c4e410c8412c23ccb9d
2023-09-15 09:54:49 -07:00
Augustin Zidek
22f6919174
Add a cast to prevent future Pandas typing issues.
...
PiperOrigin-RevId: 563001314
Change-Id: I441729e337e5bf782f70beeb92d8cf0c870bc98a
2023-09-05 23:17:42 -07:00
Augustin Zidek
c9431c4c12
Remove unused imports.
...
PiperOrigin-RevId: 562507313
Change-Id: Ic82f113455c0be50c0af1e9df4498991072e554c
2023-09-04 03:02:00 -07:00
DeepMind
bbe8a844e0
Fix: return correct renaming mask.
...
PiperOrigin-RevId: 558140222
Change-Id: I7fb69e8a7c9ce95a26d0816fb8c1c02cba68f660
2023-08-18 07:26:42 -07:00
Augustin Zidek
82ef354ca3
Add Conda lib path to LD_LIBRARY_PATH.
...
PiperOrigin-RevId: 557487629
Change-Id: I73908cf1cad254bd2bd5ec3189ac7bbae2c5f0b9
2023-08-16 08:03:22 -07:00
Hamish Tomlinson
7c9114c842
Bump jax to 0.4.14 and haiku to 0.0.10.
...
PiperOrigin-RevId: 555464931
Change-Id: I382adafe6e7f35960eeffaa1cceeaeb5e08ce87d
2023-08-10 06:28:00 -07:00
Jana Cavojska
69afc4df91
Downgrade to pyopenssl 22.0.0 to fix compatibility issue
...
PiperOrigin-RevId: 549671013
Change-Id: I8e27f86b7a02b74f8bed9437dd6d17ad8d9cabfc
2023-07-20 10:30:20 -07:00
Jana Cavojska
6cb85e9684
Update Conda in Colab and fix dependency issues
...
PiperOrigin-RevId: 549256996
Change-Id: I1c5fb68a1abe6016a89740cab1e033876c248c56
2023-07-19 02:55:00 -07:00
Dariusz Lasecki
6c4d833fbd
Add saving Protein to mmCIF file and reading Protein from mmCIF file.
...
PiperOrigin-RevId: 540873522
Change-Id: I502551e47df701cb39f72b380e9e4e818ea9c04c
2023-06-16 07:35:46 -07:00
Augustin Zidek
4bd3ff76cc
Use HTTPS instead of FTP to download all databases and switch to a faster mirror for UniRef90.
...
PiperOrigin-RevId: 536973190
Change-Id: I358b78dbeb401cd066688af69eca7b50858b69d1
2023-06-01 03:34:10 -07:00
Jake VanderPlas
9bbfdc7c81
alphafold: make compatible with JAX custom PRNG
...
This change will not modify the code's behavior, but will make the code compatible with JAX's new custom PRNG, which will soon be the default; see https://github.com/google/jax/issues/9263 .
PiperOrigin-RevId: 533069215
Change-Id: Ia98e2dd1e33e1135a628f7dc37c56b50da6c51a5
2023-05-18 03:06:00 -07:00
Jake VanderPlas
d885c55750
Future-proof alphafold prng test for jax custom PRNG
...
PRNG changes described in https://github.com/google/jax/issues/9263 . After this change, raw key buffers will no longer be accessible via indexing operations.
PiperOrigin-RevId: 530930051
Change-Id: I02b0759c528c393f7ce1109bae83da13cfd68bfe
2023-05-10 09:32:25 -07:00
DeepMind
6a3af1adb3
Rename get_pae_json function
...
PiperOrigin-RevId: 529057928
Change-Id: Idf420cf43eaa6f3e4c6cf10475cba0e06f7165ee
2023-05-03 05:03:52 -07:00
Augustin Zidek
c1e92dc524
Bump SciPy version from 1.7.0 to 1.7.3 for compatibility with Python 3.10.
...
PiperOrigin-RevId: 528849422
Change-Id: I22068400c1cd764be88e34739cbc9992d023fcda
2023-05-02 11:58:26 -07:00
Augustin Zidek
955737f12c
Upgrade OpenMM from 7.5.1 to 7.7.0 and bump Colab Python version to 3.10.
...
We don't need to patch anymore as the fix has been included upstream.
PiperOrigin-RevId: 528742992
Change-Id: I804707216bf00367c8346dffde69aeffcb7dad8f
2023-05-02 04:33:51 -07:00
DeepMind
2819de4ddd
Add pLDDT and PAE files saving to AF runner, update Colab.
...
PAE methods are migrated from Colab notebook_utils.py to AFOS and included in run_alphafold.py to save a new file. Colab notebook is updated accordingly. pLDDT data is transformed into JSON as saved as well, as an additional output file.
PiperOrigin-RevId: 525687298
Change-Id: If7f0bcf7d3b39901dae58a67958eaa5687645de2
2023-04-20 02:22:31 -07:00
Hamish Tomlinson
4d83e3fc08
Bump version to 2.3.2 in setup.py and colab.
...
PiperOrigin-RevId: 523341685
Change-Id: Ie332a4db056adb2d42bcd2f4006e4258b8908507
2023-04-11 02:23:17 -07:00