update by pingsu's advice

Signed-off-by: Future-Outlier <[email protected]>
flyteorg · Jun 4, 2024 · 40406d4 · 40406d4
1 parent 33874c0
commit 40406d4
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/flytekit/extend/backend/base_agent.py b/flytekit/extend/backend/base_agent.py
@@ -120,7 +120,7 @@ class SyncAgentBase(AgentBase):
 
     @abstractmethod
     def do(
-        self, task_template: TaskTemplate, inputs: Optional[LiteralMap], output_prefix: Optional[str], **kwargs
+        self, task_template: TaskTemplate, output_prefix: str, inputs: Optional[LiteralMap] = None, **kwargs
     ) -> Resource:
         """
         This is the method that the agent will run.
@@ -263,8 +263,8 @@ async def _do(
         self: PythonTask,
         agent: SyncAgentBase,
         template: TaskTemplate,
+        output_prefix: str,
         inputs: Dict[str, Any] = None,
-        output_prefix: Optional[str] = None,
     ) -> Resource:
         try:
             ctx = FlyteContext.current_context()

diff --git a/pyproject.toml b/pyproject.toml
@@ -23,7 +23,7 @@ dependencies = [
     "flyteidl>=1.11.0b1",
     "fsspec>=2023.3.0",
     "gcsfs>=2023.3.0",
-    "googleapis-common-protos>=1.57,!=1.63.1",
+    "googleapis-common-protos>=1.57",
     "grpcio",
     "grpcio-status",
     "importlib-metadata",