pub_soft/alphafold
3
0
Fork 0
mirror of https://github.com/google-deepmind/alphafold.git synced 2026-06-04 14:58:05 +08:00
Code Issues Packages Projects Releases Wiki Activity

Remove unused function casp_model_names.

Browse Source 
PiperOrigin-RevId: 431956393
Change-Id: I1d97992967b8c276dfa978b91bc5066efdd3b62c
This commit is contained in:
Sebastian Bodenstein
2022-03-02 09:19:51 -08:00
committed by Copybara-Service
parent 5f37a5a2ba
commit 189f7d81f3
1 changed files with 0 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
alphafold/model/data.py
View File

@@ -16,18 +16,12 @@
import io
import os
from typing import List
from alphafold.model import utils
import haiku as hk
import numpy as np
# Internal import (7716).
def casp_model_names(data_dir: str) -> List[str]:
params = os.listdir(os.path.join(data_dir, 'params'))
return [os.path.splitext(filename)[0] for filename in params]
def get_model_haiku_params(model_name: str, data_dir: str) -> hk.Params:
"""Get the Haiku parameters from a model name."""