From ac5591f6f3aa685e20f9187a8dea10ba0e57a32e Mon Sep 17 00:00:00 2001
From: Brennan Abanades Kenyon <56822523+brennanaba@users.noreply.github.com>
Date: Thu, 10 Nov 2022 15:27:43 +0000
Subject: [PATCH] Update models.py

---
 ImmuneBuilder/models.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/ImmuneBuilder/models.py b/ImmuneBuilder/models.py
index 9f58050..b4f4156 100644
--- a/ImmuneBuilder/models.py
+++ b/ImmuneBuilder/models.py
@@ -225,7 +225,7 @@ def forward(self, node_features, sequence):
 
         # Remove atoms of side chains with outrageous clashes
         ds = torch.linalg.norm(all_atoms[None,:,None] - all_atoms[:,None,:,None], axis = -1)
-        ds[torch.isnan(ds!=ds) | (ds==0.0)] = 10
+        ds[torch.isnan(ds) | (ds==0.0)] = 10
         min_ds = ds.min(dim=-1)[0].min(dim=-1)[0].min(dim=-1)[0]
         all_atoms[min_ds < 0.2, 5:, :] = float("Nan")