Commit 1d657f55 authored by Gustaf Ahdritz's avatar Gustaf Ahdritz
Browse files

Appease flake8 around importlib import

parent 94d890d8
......@@ -48,7 +48,7 @@ def import_alphafold():
def get_alphafold_config():
config = alphafold.model.config.model_config("model_1_ptm")
config = alphafold.model.config.model_config("model_1_ptm") # noqa
config.model.global_config.deterministic = True
return config
......@@ -102,7 +102,7 @@ def fetch_alphafold_module_weights(weight_path):
_remove_key_prefix(params, prefix)
try:
params = alphafold.model.utils.flat_params_to_haiku(params)
params = alphafold.model.utils.flat_params_to_haiku(params) # noqa
except:
raise ImportError(
"Make sure to call import_alphafold before running this function"
......
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