fix typo

EleutherAI · Jul 22, 2021 · 2ce9300 · 2ce9300
1 parent 854bafe
commit 2ce9300
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/mp_nerf/ml_utils.py b/mp_nerf/ml_utils.py
@@ -99,7 +99,7 @@ def torsion_angle_loss(pred_torsions, true_torsions, coeff=2., angle_mask=None):
     return coeff * (1 - maxi)
 
 
-def fape_torch(pred_coords, true_coords, max_val=10., l_func=None
+def fape_torch(pred_coords, true_coords, max_val=10., l_func=None,
                c_alpha=False, seq_list=None, rot_mats_g=None): 
     """ Computes the Frame-Aligned Point Error. Scaled 0 <= FAPE <= 1
         Inputs: