Skip to content

GitLab

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

Appease flake8 around importlib import

parent 94d890d8
Show whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
tests/compare_utils.py
View file @ 1d657f55
...@@ -48,7 +48,7 @@ def import_alphafold(): ...@@ -48,7 +48,7 @@ def import_alphafold():
def get_alphafold_config(): 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 config.model.global_config.deterministic = True
return config return config
...@@ -102,7 +102,7 @@ def fetch_alphafold_module_weights(weight_path): ...@@ -102,7 +102,7 @@ def fetch_alphafold_module_weights(weight_path):
_remove_key_prefix(params, prefix) _remove_key_prefix(params, prefix)
try: try:
params = alphafold.model.utils.flat_params_to_haiku(params) params = alphafold.model.utils.flat_params_to_haiku(params) # noqa
except: except:
raise ImportError( raise ImportError(
"Make sure to call import_alphafold before running this function" "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