Commit 189f7d81 authored by Sebastian Bodenstein's avatar Sebastian Bodenstein Committed by Copybara-Service
Browse files

Remove unused function casp_model_names.

PiperOrigin-RevId: 431956393
Change-Id: I1d97992967b8c276dfa978b91bc5066efdd3b62c
parent 5f37a5a2
......@@ -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."""
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment