fix done/adone test

pygod-team · May 11, 2023 · d049fcd · d049fcd
1 parent 16b8247
commit d049fcd
Show file tree

Hide file tree

Showing 4 changed files with 4 additions and 4 deletions.
diff --git a/pygod/detector/adone.py b/pygod/detector/adone.py
@@ -187,7 +187,7 @@ def init_model(self, **kwargs):
 
     def forward_model(self, data):
         batch_size = data.batch_size
-        node_idx = data.node_idx
+        node_idx = data.n_id
 
         x = data.x.to(self.device)
         s = data.s.to(self.device)

diff --git a/pygod/detector/anomalydae.py b/pygod/detector/anomalydae.py
@@ -171,7 +171,7 @@ def init_model(self, **kwargs):
 
     def forward_model(self, data):
         batch_size = data.batch_size
-        node_idx = data.node_idx
+        node_idx = data.n_id
 
         x = data.x.to(self.device)
         s = data.s.to(self.device)

diff --git a/pygod/detector/conad.py b/pygod/detector/conad.py
@@ -186,7 +186,7 @@ def init_model(self, **kwargs):
 
     def forward_model(self, data):
         batch_size = data.batch_size
-        node_idx = data.node_idx
+        node_idx = data.n_id
 
         x = data.x.to(self.device)
         s = data.s.to(self.device)

diff --git a/pygod/detector/done.py b/pygod/detector/done.py
@@ -186,7 +186,7 @@ def init_model(self, **kwargs):
 
     def forward_model(self, data):
         batch_size = data.batch_size
-        node_idx = data.node_idx
+        node_idx = data.n_id
 
         x = data.x.to(self.device)
         s = data.s.to(self.device)