Make parsers printable

yzhangcs · May 24, 2023 · 6f0f60b · 6f0f60b
1 parent ab9ba00
commit 6f0f60b
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/supar/parser.py b/supar/parser.py
@@ -44,6 +44,12 @@ def __init__(self, args, model, transform):
         self.model = model
         self.transform = transform
 
+    def __repr__(self):
+        s = f'{self.__class__.__name__}(\n'
+        s += '\n'.join(['  '+i for i in str(self.model).split('\n')]) + '\n'
+        s += '\n'.join(['  '+i for i in str(self.transform).split('\n')]) + '\n)'
+        return s
+
     @property
     def device(self):
         return 'cuda' if torch.cuda.is_available() else 'cpu'