Update ProteinMPNN/vanilla_proteinmpnn/protein_mpnn_utils.py

ProteinMPNN/vanilla_proteinmpnn/protein_mpnn_utils.py

@@ -734,7 +734,7 @@ class ProteinFeatures(nn.Module):
 
     def forward(self, X, mask, residue_idx, chain_labels):
         if self.augment_eps !='':
-            if int(self.augment_eps) > 0:
+            if self.augment_eps > 0:
                 X = X + self.augment_eps * torch.randn_like(X)
         
         b = X[:,:,1,:] - X[:,:,0,:]