From f29dbfe4aa3b2fba2c399edf6e6867028b2739b1 Mon Sep 17 00:00:00 2001
From: Jeewanghimire31 <jeewanghimire31@gmail.com>
Date: Sat, 28 Dec 2024 12:17:32 +0545
Subject: [PATCH] fix/typo: Fix typo issue on file import

---
 examples/examples.py      | 7 ++++---
 pygip/protect/__init__.py | 2 +-
 2 files changed, 5 insertions(+), 4 deletions(-)

diff --git a/examples/examples.py b/examples/examples.py
index 6509d67..5137a33 100644
--- a/examples/examples.py
+++ b/examples/examples.py
@@ -1,10 +1,10 @@
-import sys
 import os
+import sys
+
 sys.path.append(os.path.abspath(os.path.join(os.path.dirname(__file__), '..')))
 from pygip.datasets.datasets import *
 from pygip.protect import *
-from pygip.protect.Defense import Watermark_sage
-
+from pygip.protect.defense import Watermark_sage
 
 # dataset = Cora()
 # dataset = Citeseer()
@@ -135,3 +135,4 @@ def test():
         elif (dataset_name == 3):
             defense = Watermark_sage(PubMed(), 0.25)
             defense.watermark_attack(PubMed(), attack_name, dataset_name)
+test()
\ No newline at end of file
diff --git a/pygip/protect/__init__.py b/pygip/protect/__init__.py
index 733ba09..1846720 100644
--- a/pygip/protect/__init__.py
+++ b/pygip/protect/__init__.py
@@ -1 +1 @@
-from .gnn_mea import *
+from .attack import *