Skip to content

Commit 51c6765

Browse files
committed
Move aiperf code to top-level
1 parent 967c6ce commit 51c6765

File tree

8 files changed

+4
-4
lines changed

8 files changed

+4
-4
lines changed
File renamed without changes.

nemoguardrails/benchmark/aiperf/__init__.py renamed to aiperf/__init__.py

File renamed without changes.

nemoguardrails/benchmark/aiperf/aiperf_configs/single_concurrency.yaml renamed to aiperf/aiperf_configs/single_concurrency.yaml

File renamed without changes.

nemoguardrails/benchmark/aiperf/aiperf_configs/sweep_concurrency.yaml renamed to aiperf/aiperf_configs/sweep_concurrency.yaml

File renamed without changes.

nemoguardrails/benchmark/aiperf/aiperf_models.py renamed to aiperf/aiperf_models.py

File renamed without changes.

nemoguardrails/benchmark/aiperf/run_aiperf.py renamed to aiperf/run_aiperf.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -32,7 +32,7 @@
3232
import yaml
3333
from pydantic import ValidationError
3434

35-
from nemoguardrails.benchmark.aiperf.aiperf_models import AIPerfConfig
35+
from aiperf.aiperf_models import AIPerfConfig
3636

3737
log = logging.getLogger(__name__)
3838
log.setLevel(logging.INFO)

nemoguardrails/cli/__init__.py

Lines changed: 0 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -25,7 +25,6 @@
2525

2626
from nemoguardrails import __version__
2727
from nemoguardrails.actions_server import actions_server
28-
from nemoguardrails.benchmark.aiperf.run_aiperf import app as aiperf_app
2928
from nemoguardrails.cli.chat import run_chat
3029
from nemoguardrails.cli.migration import migrate
3130
from nemoguardrails.cli.providers import _list_providers, select_provider_with_type
@@ -47,7 +46,6 @@ class ColangVersions(str, Enum):
4746
app.add_typer(
4847
eval_cli.app, name="eval", short_help="Evaluation a guardrail configuration."
4948
)
50-
app.add_typer(aiperf_app, name="aiperf", short_help="Run AIPerf benchmarks")
5149
app.pretty_exceptions_enable = False
5250

5351
logging.getLogger().setLevel(logging.WARNING)

nemoguardrails/llm/providers/huggingface/streamers.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -18,7 +18,9 @@
1818

1919
TRANSFORMERS_AVAILABLE = True
2020
try:
21-
from transformers.generation.streamers import TextStreamer # type: ignore
21+
from transformers.generation.streamers import ( # type: ignore[import-untyped]
22+
TextStreamer,
23+
)
2224
except ImportError:
2325
# Fallback if transformers is not available
2426
TRANSFORMERS_AVAILABLE = False

0 commit comments

Comments
 (0)