Merge branch 'partitioner' of https://github.com/open-sciencelab/GraphGen into partitioner

Author: ChenZiHong-Gavin

graphgen/models/llm/openai_client.py

@@ -39,8 +39,8 @@ def __init__(
         seed: Optional[int] = None,
         topk_per_token: int = 5,  # number of topk tokens to generate for each token
         request_limit: bool = False,
-        rpm: RPM = None,
-        tpm: TPM = None,
+        rpm: Optional[RPM] = None,
+        tpm: Optional[TPM] = None,
         **kwargs: Any,
     ):
         super().__init__(**kwargs)