style(ops): type-annotate Ops & builder clases

Author: ankostis

graphtik/netop.py

@@ -42,7 +42,7 @@ def _build_pydot(self, **kws):
         plotter = self.last_plan or self.net
         return plotter._build_pydot(**kws)
 
-    def compute(self, named_inputs, outputs=None):
+    def compute(self, named_inputs, outputs=None) -> dict:
         """
         Solve & execute the graph, sequentially or parallel.
 
@@ -82,7 +82,7 @@ def compute(self, named_inputs, outputs=None):
 
     def __call__(
         self, named_inputs, outputs=None, method=None, overwrites_collector=None
-    ):
+    ) -> dict:
         return self.compute(named_inputs, outputs=outputs)
 
     def set_execution_method(self, method):
@@ -145,7 +145,7 @@ def __init__(self, name=None, merge=False):
         self.name = name
         self.merge = merge
 
-    def __call__(self, *operations, **kwargs):
+    def __call__(self, *operations, **kwargs) -> NetworkOperation:
         """
         Composes a collection of operations into a single computation graph,
         obeying the ``merge`` property, if set in the constructor.

graphtik/op.py

@@ -128,7 +128,7 @@ def __repr__(self):
             f"provides={provides!r}, fn{returns_dict_marker}={fn_name!r})"
         )
 
-    def compute(self, named_inputs, outputs=None):
+    def compute(self, named_inputs, outputs=None) -> dict:
         try:
             args = [
                 named_inputs[n]
@@ -239,7 +239,7 @@ def __init__(
 
     def withset(
         self, *, fn=None, name=None, needs=None, provides=None, returns_dict=None
-    ):
+    ) -> "operation":
         if fn is not None:
             self.fn = fn
         if name is not None:
@@ -255,7 +255,7 @@ def withset(
 
     def __call__(
         self, fn=None, *, name=None, needs=None, provides=None, returns_dict=None
-    ):
+    ) -> FunctionalOperation:
         """
         This enables ``operation`` to act as a decorator or as a functional
         operation, for example::