remove types

jxnl · Jul 22, 2023 · 84424a4 · 84424a4
1 parent 9802a74
commit 84424a4
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/openai_function_call/dsl/multitask.py b/openai_function_call/dsl/multitask.py
@@ -3,7 +3,7 @@
 from openai_function_call import OpenAISchema
 
 
-class MultiTaskBase(OpenAISchema):
+class MultiTaskBase:
     task_type = None  # type: ignore
 
     @classmethod
@@ -51,7 +51,7 @@ def MultiTask(
     subtask_class: Type[BaseModel],
     name: Optional[str] = None,
     description: Optional[str] = None,
-) -> Union[MultiTaskBase | OpenAISchema]:
+):
     """
     Dynamically create a MultiTask OpenAISchema that can be used to segment multiple
     tasks given a base class. This creates class that can be used to create a toolkit