Mercurial > repos > galaxy-australia > alphafold2
view docker/alphafold/alphafold/model/data.py @ 1:6c92e000d684 draft
"planemo upload for repository https://github.com/usegalaxy-au/galaxy-local-tools commit a510e97ebd604a5e30b1f16e5031f62074f23e86"
author | galaxy-australia |
---|---|
date | Tue, 01 Mar 2022 02:53:05 +0000 |
parents | |
children |
line wrap: on
line source
# Copyright 2021 DeepMind Technologies Limited # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. # You may obtain a copy of the License at # # http://www.apache.org/licenses/LICENSE-2.0 # # Unless required by applicable law or agreed to in writing, software # distributed under the License is distributed on an "AS IS" BASIS, # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. # See the License for the specific language governing permissions and # limitations under the License. """Convenience functions for reading data.""" 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.""" path = os.path.join(data_dir, 'params', f'params_{model_name}.npz') with open(path, 'rb') as f: params = np.load(io.BytesIO(f.read()), allow_pickle=False) return utils.flat_params_to_haiku(params)