diff --git a/src/palimpzest/query/generators/generators.py b/src/palimpzest/query/generators/generators.py index e7d8e1ac..9ad5c9b1 100644 --- a/src/palimpzest/query/generators/generators.py +++ b/src/palimpzest/query/generators/generators.py @@ -12,7 +12,7 @@ from abc import ABC, abstractmethod from collections import Counter from string import Formatter -from typing import Any, Generic, Tuple, TypeVar +from typing import Any, Generic, TypeVar from colorama import Fore, Style from openai import OpenAI @@ -40,7 +40,7 @@ from palimpzest.utils.sandbox import API # DEFINITIONS -GenerationOutput = Tuple[dict, str | None, GenerationStats] +GenerationOutput = tuple[dict, str | None, GenerationStats] ContextType = TypeVar("ContextType") InputType = TypeVar("InputType") diff --git a/src/palimpzest/query/operators/code_synthesis_convert.py b/src/palimpzest/query/operators/code_synthesis_convert.py index 5e9d610f..650fc613 100644 --- a/src/palimpzest/query/operators/code_synthesis_convert.py +++ b/src/palimpzest/query/operators/code_synthesis_convert.py @@ -1,6 +1,6 @@ from __future__ import annotations -from typing import Any, Dict, Tuple +from typing import Any from palimpzest.constants import Cardinality, GPT_4o_MODEL_CARD, Model from palimpzest.core.data.dataclasses import GenerationStats, OperatorCostEstimates @@ -15,9 +15,9 @@ FieldName = str CodeName = str Code = str -DataRecordDict = Dict[str, Any] -Exemplar = Tuple[DataRecordDict, DataRecordDict] -CodeEnsemble = Dict[CodeName, Code] +DataRecordDict = dict[str, Any] +Exemplar = tuple[DataRecordDict, DataRecordDict] +CodeEnsemble = dict[CodeName, Code] class CodeSynthesisConvert(LLMConvert): diff --git a/src/palimpzest/query/operators/logical.py b/src/palimpzest/query/operators/logical.py index b62c7ce8..4bb739de 100644 --- a/src/palimpzest/query/operators/logical.py +++ b/src/palimpzest/query/operators/logical.py @@ -137,11 +137,7 @@ def get_logical_op_params(self) -> dict: class BaseScan(LogicalOperator): """A BaseScan is a logical operator that represents a scan of a particular data source.""" - def __init__( - self, - dataset_id: str, - output_schema: Schema, - ): + def __init__(self, dataset_id: str, output_schema: Schema): super().__init__(output_schema=output_schema) self.dataset_id = dataset_id