mypy

Author: tymekaxai

src/axiomatic/axtract/interactive_table.py

@@ -3,6 +3,7 @@
 import json  # type: ignore
 import os  # type: ignore
 from .. import EquationProcessingResponse, VariableRequirement
+from typing import Dict, Any
 
 
 def _find_symbol(name, variable_dict):
@@ -155,7 +156,7 @@ def save_custom_presets(custom_data, file_path):
     name_label_width = ["150px"]
 
     # Dictionary to keep track of row widget references
-    value_widgets = {}
+    value_widgets: Dict[str, Any] = {}
 
     # ---------------------------------------------------------------
     # 6) display_table(change): Re-populate rows when user selects a template

src/axiomatic/client.py

@@ -5,7 +5,7 @@
 import os
 import time
 import json
-from typing import Dict, List, Optional, Union
+from typing import Dict, List, Optional, Union, Sequence, Any
 
 from .base_client import BaseClient, AsyncBaseClient
 from . import ParseResponse, EquationProcessingResponse
@@ -116,7 +116,7 @@ def analyze_equations(
 
     def validate_equations(
         self,
-        requirements: list[VariableRequirement],
+        requirements: List[VariableRequirement],
         loaded_equations: EquationProcessingResponse,
         include_internal_model: bool = False,
     ) -> EquationValidationResult:

src/axiomatic/types/equation_processing.py

@@ -1,5 +1,5 @@
 from pydantic import BaseModel, Field
-from typing import List, Dict, Optional
+from typing import List, Dict, Optional, Any
 
 class DictItem(BaseModel):
     key: str
@@ -19,20 +19,20 @@ class ResponseEquation(BaseModel):
     wolfram_expressions: str = Field(
         ..., description="Equation in Wolfram Language (or other symbolic form)"
     )
-    latex_symbols: list[DictItem] = Field(
+    latex_symbols: List[DictItem] = Field(
         ..., description="Detailed metadata for each latex variable in original format"
     )
-    wolfram_symbols: list[str] = Field(
+    wolfram_symbols: List[str] = Field(
         ..., description="List of symbols used in the wolfram expression"
     )
-    narrative_assumptions: list[str] = Field(
+    narrative_assumptions: List[str] = Field(
         ..., description="Narrative text describing assumptions/approximations"
     )
-    type: list[str] = Field(
+    type: List[str] = Field(
         ...,
         description="List of equation classifications (e.g., ['scalar','tensor'])",
     )
-    field_tags: list[str] = Field(
+    field_tags: List[str] = Field(
         ...,
         description="Classification tags (e.g., ['quantum mechanics','gravitation'])",
     )

src/axiomatic/types/validate_equations_body.py

@@ -1,8 +1,9 @@
 from .variable_requirement import VariableRequirement
 from .equation_processing_response import EquationProcessingResponse
 from pydantic import BaseModel
+from typing import List
 
 class ValidateEquationsBody(BaseModel):
-    variables: list[VariableRequirement]
+    variables: List[VariableRequirement]
     paper_equations: EquationProcessingResponse
     include_internal_model: bool = False