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 @@ ...@@ -16,18 +16,12 @@
import io import io
import os import os
from typing import List
from alphafold.model import utils from alphafold.model import utils
import haiku as hk import haiku as hk
import numpy as np import numpy as np
# Internal import (7716). # 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: def get_model_haiku_params(model_name: str, data_dir: str) -> hk.Params:
"""Get the Haiku parameters from a model name.""" """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