Skip to content

style: fix return type hints #512

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
shuangwu5 merged 1 commit into from
Jul 3, 2025
Merged

style: fix return type hints #512

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
1 commit
Select commit
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
2 changes: 1 addition & 1 deletion mostlyai/sdk/client/_base_utils.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -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")):
Expand Down
6 changes: 3 additions & 3 deletions mostlyai/sdk/client/generators.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -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"],
Expand Down Expand Up @@ -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"],
Expand Down Expand Up @@ -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"],
Expand Down
8 changes: 5 additions & 3 deletions mostlyai/sdk/client/synthetic_datasets.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -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"],
Expand Down Expand Up @@ -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"],
Expand Down Expand Up @@ -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"],
Expand Down