Merge pull request #8 from yahoo/informative_names

Informative names
yahoo · Dec 15, 2017 · 708eb1f · 708eb1f
2 parents d7039f8 + dcd3001
commit 708eb1f
Show file tree

Hide file tree

Showing 2 changed files with 23 additions and 0 deletions.
diff --git a/graphkit/base.py b/graphkit/base.py
@@ -133,6 +133,16 @@ def __setstate__(self, state):
             self.__setattr__(k, state[k])
         self._after_init()
 
+    def __repr__(self):
+        """
+        Display more informative names for the Operation class
+        """
+        return u"%s(name='%s', needs=%s, provides=%s)" % \
+            (self.__class__.__name__,
+             self.name,
+             self.needs,
+             self.provides)
+
 
 class NetworkOperation(Operation):
     def __init__(self, **kwargs):

diff --git a/graphkit/functional.py b/graphkit/functional.py
@@ -128,6 +128,19 @@ def myadd(a, b):
 
         return FunctionalOperation(**total_kwargs)
 
+    def __repr__(self):
+        """
+        Display more informative names for the Operation class
+        """
+        return u"%s(name='%s', needs=%s, provides=%s, fn=%s)" % \
+            (self.__class__.__name__,
+             self.name,
+             self.needs,
+             self.provides,
+             self.fn.__name__)
+
+
+
 
 class compose(object):
     """