diff --git a/mostlyai/sdk/client/_base_utils.py b/mostlyai/sdk/client/_base_utils.py index c3826ac0..99e6d243 100644 --- a/mostlyai/sdk/client/_base_utils.py +++ b/mostlyai/sdk/client/_base_utils.py @@ -72,7 +72,7 @@ def convert_to_df(data: str, format: Literal["parquet", "jsonl"] = "parquet") -> return df -def read_table_from_path(path: str | Path) -> (str, pd.DataFrame): +def read_table_from_path(path: str | Path) -> tuple[str, pd.DataFrame]: # read data from file fn = str(path) if fn.lower().endswith((".pqt", ".parquet")): diff --git a/mostlyai/sdk/client/generators.py b/mostlyai/sdk/client/generators.py index f9ef1739..8ae2ffc7 100644 --- a/mostlyai/sdk/client/generators.py +++ b/mostlyai/sdk/client/generators.py @@ -267,7 +267,7 @@ def import_from_file( def _export_to_file( self, generator_id: str, - ) -> (bytes, str | None): + ) -> tuple[bytes, str | None]: response = self.request( verb=GET, path=[generator_id, "export-to-file"], @@ -318,7 +318,7 @@ def _report( source_table_id: str, model_type: ModelType = ModelType.tabular, short_lived_file_token: str | None = None, - ) -> (str, str | None): + ) -> tuple[str, str | None]: response = self.request( verb=GET, path=[generator_id, "tables", source_table_id, "report"], @@ -348,7 +348,7 @@ def _training_wait(self, generator_id: str, progress_bar: bool, interval: float) generator = self.get(generator_id) return generator - def _training_logs(self, generator_id: str, short_lived_file_token: str | None = None) -> (bytes, str): + def _training_logs(self, generator_id: str, short_lived_file_token: str | None = None) -> tuple[bytes, str]: response = self.request( verb=GET, path=[generator_id, "training", "logs"], diff --git a/mostlyai/sdk/client/synthetic_datasets.py b/mostlyai/sdk/client/synthetic_datasets.py index 5a44dfb3..ed5bcf7c 100644 --- a/mostlyai/sdk/client/synthetic_datasets.py +++ b/mostlyai/sdk/client/synthetic_datasets.py @@ -213,7 +213,7 @@ def _download( synthetic_dataset_id: str, ds_format: SyntheticDatasetFormat = SyntheticDatasetFormat.parquet, short_lived_file_token: str | None = None, - ) -> (bytes, str | None): + ) -> tuple[bytes, str | None]: response = self.request( verb=GET, path=[synthetic_dataset_id, "download"], @@ -261,7 +261,7 @@ def _report( model_type: ModelType = ModelType.tabular, report_type: SyntheticDatasetReportType = SyntheticDatasetReportType.model, short_lived_file_token: str | None = None, - ) -> (str, str | None): + ) -> tuple[str, str | None]: response = self.request( verb=GET, path=[synthetic_dataset_id, "tables", synthetic_table_id, "report"], @@ -300,7 +300,9 @@ def _generation_wait(self, synthetic_dataset_id: str, progress_bar: bool, interv synthetic_dataset = self.get(synthetic_dataset_id) return synthetic_dataset - def _generation_logs(self, synthetic_dataset_id: str, short_lived_file_token: str | None = None) -> (bytes, str): + def _generation_logs( + self, synthetic_dataset_id: str, short_lived_file_token: str | None = None + ) -> tuple[bytes, str]: response = self.request( verb=GET, path=[synthetic_dataset_id, "generation", "logs"],