mypy unpack ignore statements

Author: Andrei Bratu

.github/workflows/ci.yml

@@ -17,7 +17,7 @@ jobs:
       - name: Install dependencies
         run: poetry install
       - name: Compile
-        run: poetry run mypy . --enable-incomplete-feature=Unpack
+        run: poetry run mypy .
   test:
     runs-on: ubuntu-20.04
     steps:

pyproject.toml

@@ -54,7 +54,7 @@ cohere = "^5.11.2"
 replicate = "^1.0.3"
 jsonschema = "^4.23.0"
 types-jsonschema = "^4.23.0.20240813"
-mypy = "^1.6.0"
+mypy = "^1.0.1"
 pytest = "^7.4.0"
 pytest-asyncio = "^0.23.5"
 python-dateutil = "^2.8.2"

src/humanloop/decorators/flow.py

@@ -20,7 +20,7 @@
 def flow(
     opentelemetry_tracer: Tracer,
     path: Optional[str] = None,
-    **flow_kernel: Unpack[FlowKernelRequestParams],
+    **flow_kernel: Unpack[FlowKernelRequestParams],  # type: ignore
 ):
     flow_kernel["attributes"] = {k: v for k, v in flow_kernel.get("attributes", {}).items() if v is not None}
 

src/humanloop/decorators/prompt.py

@@ -20,7 +20,7 @@ def prompt(
     opentelemetry_tracer: Tracer,
     path: Optional[str] = None,
     # TODO: Template can be a list of objects?
-    **prompt_kernel: Unpack[DecoratorPromptKernelRequestParams],
+    **prompt_kernel: Unpack[DecoratorPromptKernelRequestParams],  # type: ignore
 ):
     def decorator(func: Callable):
         @wraps(func)

src/humanloop/decorators/tool.py

@@ -30,7 +30,7 @@
 def tool(
     opentelemetry_tracer: Tracer,
     path: Optional[str] = None,
-    **tool_kernel: Unpack[ToolKernelRequestParams],
+    **tool_kernel: Unpack[ToolKernelRequestParams],  # type: ignore
 ):
     def decorator(func: Callable):
         enhanced_tool_kernel = _build_tool_kernel(