Rebase fixes

Author: Andrei Bratu

src/humanloop/eval_utils/run.py

@@ -111,7 +111,7 @@ def _is_evaluated_file(
         ) == log_args.get("path")
 
     # Copy the original log method in a hidden attribute
-    client._log = client.log
+    client._log = client.log  # type: ignore
 
     def _overloaded_log(
         self,

src/humanloop/evaluations/client.py

@@ -14,7 +14,6 @@
 from .requests.create_evaluation_request_evaluators_item import (
     CreateEvaluationRequestEvaluatorsItemParams,
 )
-from ..requests.evaluations_request import EvaluationsRequestParams
 from ..requests.file_request import FileRequestParams
 from ..core.serialization import convert_and_respect_annotation_metadata
 from .requests.add_evaluators_request_evaluators_item import (
@@ -26,11 +25,6 @@
 from .requests.create_run_request_version import CreateRunRequestVersionParams
 from ..types.evaluation_run_response import EvaluationRunResponse
 from ..types.evaluation_status import EvaluationStatus
-from ..requests.evaluations_dataset_request import EvaluationsDatasetRequestParams
-from ..requests.version_specification import VersionSpecificationParams
-from ..types.logs_association_type import LogsAssociationType
-from ..types.evaluation_run_response import EvaluationRunResponse
-from ..types.evaluation_status import EvaluationStatus
 from ..types.evaluation_stats import EvaluationStats
 from ..types.paginated_data_evaluation_log_response import (
     PaginatedDataEvaluationLogResponse,

src/humanloop/otel/helpers.py

@@ -105,7 +105,7 @@ def write_to_opentelemetry_span(
             # OTel does not allow lists of complex objects, so we linearise them
             # by mapping each dict to an index key and recursing into the dict
             for idx, list_value in enumerate(value):
-                work_stack.append((f"{key}.{idx}" if key else idx, list_value))
+                work_stack.append((f"{key}.{idx}" if key else idx, list_value))  # type: ignore
         else:
             linearised_attributes[key] = value  # type: ignore
     for final_key, final_value in linearised_attributes.items():
@@ -232,12 +232,13 @@ def pseudo_to_list(sub_dict):
     result = pseudo_to_list(result)
     if "" in result:
         # User read the root of attributes
-        return result[""]
+        return result[""]  # type: ignore
 
     for part in key.split("."):
         if str.isnumeric(part):
-            part = int(part)
-        result = result[part]  # type: ignore
+            result = result[int(part)]  # type: ignore
+        else:
+            result = result[part]  # type: ignore
 
     return result