Tmp commit

src/omnipy/compute/task.py

@@ -61,6 +61,38 @@ def a(a: int) -> int:
 #
 # TaskTemplate = get_task_template()
 
+
+class ListDoubler(Generic[C]):
+    def __init__(self, l: list[C]) -> None:
+        self._l = list
+
+    def get_double_l(self) -> list[C]:
+        return self._l + self._l
+
+
+reveal_type(ListDoubler)
+reveal_type(TaskTemplateInner)
+
+
+class IsPrintingListDoubler(Protocol[C]):
+    def get_double_l(self):
+        ...
+
+    def print(self):
+        ...
+
+
+def add_print(cls: Type[ListDoubler[C]]) -> IsPrintingListDoubler[C]:
+    def _inner():
+        print(cls.get_double_l())
+
+    cls.print = _inner
+    return cast(IsPrintingListDoubler[C], cls)
+
+
+P = add_print(ListDoubler)
+reveal_type(P)
+
 TaskTemplate = task_template_callable_decorator_cls(TaskTemplateInner)
 reveal_type(TaskTemplate)
 

src/omnipy/modules/fairtracks/tasks.py

@@ -21,3 +21,13 @@ def import_dataset_from_encode(endpoints: Iterable[constr(min_length=1)],
             format='json',
         )
     return dataset
+
+
+def a(a: int) -> int:
+    return a + 2
+
+
+tttts = TaskTemplate()(a)
+reveal_type(tttts)
+b = tttts.run(a=2)
+reveal_type(b)