Update base.py

Small change for the paper publication.
muammar · Feb 11, 2020 · 859e832 · 859e832
1 parent 495539f
commit 859e832
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/ml4chem/models/base.py b/ml4chem/models/base.py
@@ -6,7 +6,7 @@ class DeepLearningModel(ABC, torch.nn.Module):
     @abstractmethod
     def name(cls):
         """Return name of the class"""
-        pass
+        return cls.NAME
 
     @abstractmethod
     def __init__(self, **kwargs):