NousResearch/atropos
1
0
Fork 0
mirror of https://github.com/NousResearch/atropos.git synced 2026-04-19 12:57:58 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

python versioning problems

Browse source
This commit is contained in:
Jai Suphavadeeprasit 2026-02-03 09:55:22 -05:00
parent 4f6f2faac8
commit 00649296a1
1 changed files with 11 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

13
example_trainer/vllm_api_server.py
View file

@ -118,8 +118,17 @@ from vllm.v1.engine.async_llm import AsyncLLM # noqa: E402
try:
from vllm.utils import FlexibleArgumentParser
except ImportError:
# Use standard argparse for newer vLLM versions
from argparse import ArgumentParser as FlexibleArgumentParser
# Create a compatible ArgumentParser that handles 'deprecated' kwarg
# (Python 3.10 doesn't support 'deprecated' in BooleanOptionalAction)
import argparse
class FlexibleArgumentParser(argparse.ArgumentParser):
"""ArgumentParser that strips unsupported kwargs for Python < 3.13."""
def add_argument(self, *args, **kwargs):
# Remove 'deprecated' kwarg if present (not supported before Python 3.13)
kwargs.pop('deprecated', None)
return super().add_argument(*args, **kwargs)
# set_ulimit might not exist in all vLLM versions
try: