diff --git a/dataset.py b/dataset.py index 2222196..f8778a6 100644 --- a/dataset.py +++ b/dataset.py @@ -1,6 +1,6 @@ from dataclasses import dataclass, field from enum import Enum -from typing import Any, Dict, List, Optional +from typing import Any, Dict, List, Literal, Optional, Union import json, argparse, os, uuid from utils import prompt_yes_no @@ -124,7 +124,7 @@ class Dataset: to_dump = Dataset(entries=entries_to_dump) print(f"{len(entries_to_dump)} entries...", end=" ", flush=True) - def transform_entry(entry: DatasetEntry | Dataset | Any) -> dict | list: + def transform_entry(entry: Union[DatasetEntry, Dataset, Any]) -> Union[dict, list]: if not isinstance(entry, (DatasetEntry, Dataset)): return entry.__dict__ diff --git a/utils.py b/utils.py index da841b5..6c5b4d6 100644 --- a/utils.py +++ b/utils.py @@ -1,5 +1,6 @@ import os, sys, logging, subprocess from datetime import datetime +from typing import Optional from github.Commit import Commit from github.PaginatedList import PaginatedList from github.PullRequestComment import PullRequestComment @@ -166,7 +167,7 @@ def run_git_cmd(cmd: list[str], repo_path: str) -> subprocess.CompletedProcess: ) -def prompt_yes_no(prompt: str, *, default: bool | None = None) -> bool: +def prompt_yes_no(prompt: str, *, default: Optional[bool] = None) -> bool: choices = "y/n" if default is not None: choices = "Y/n" if default else "y/N"