Fix formatting

91f031f5 · Gustaf Ahdritz · b1780f4e · 91f031f5
Commit 91f031f5 authored Jul 03, 2022 by Gustaf Ahdritz
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 2 deletions

openfold/utils/import_weights.py openfold/utils/import_weights.py +2 -2

No files found.
--- a/openfold/utils/import_weights.py
+++ b/openfold/utils/import_weights.py
@@ -397,8 +397,7 @@ def generate_translation_dict(model, version):
        tps_blocks = model.template_pair_stack.blocks
        tps_blocks_params = stacked(
            [TemplatePairBlockParams(b) for b in tps_blocks]
-        )
-        
+        ) 
        template_param_dict = {
            "template_embedding": {
                "single_template_embedding": {
@@ -421,6 +420,7 @@ def generate_translation_dict(model, version):
                model.template_angle_embedder.linear_2
            ),
        }
+        
        translations["evoformer"].update(template_param_dict)   

    if "_ptm" in version: