Merge pull request #52 from gizatechxyz/fix/model-category-agents

Fix XGB inference inside Agents
gizatechxyz · May 23, 2024 · 31d91f6 · 31d91f6
2 parents eff9bd9 + 99b50a8
commit 31d91f6
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/giza/agents/agent.py b/giza/agents/agent.py
@@ -252,6 +252,7 @@ def predict(
         custom_output_dtype: Optional[str] = None,
         job_size: str = "M",
         dry_run: bool = False,
+        model_category: Optional[str] = None,
         **result_kwargs: Any,
     ) -> Optional[Union[Tuple[Any, Any], "AgentResult"]]:
         """
@@ -270,6 +271,7 @@ def predict(
             custom_output_dtype=custom_output_dtype,
             job_size=job_size,
             dry_run=dry_run,
+            model_category=model_category,
         )
 
         self.verifiable = verifiable

diff --git a/pyproject.toml b/pyproject.toml
@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "giza-agents"
-version = "0.4.0"
+version = "0.4.1"
 
 description = "A Python SDK for Giza platform"
 authors = [