Skip to content

Commit

Permalink
[BugFix] Hide sparseml.benchmark (#1625)
Browse files Browse the repository at this point in the history
* Hides `sparseml.benchmark` utility for now, scheduled for deprecation in the future

* Address review comments from @bfineran

---------

Co-authored-by: Benjamin Fineran <bfineran@users.noreply.github.com>
  • Loading branch information
rahul-tuli and bfineran committed Jun 26, 2023
1 parent 5a3697f commit 137abc0
Show file tree
Hide file tree
Showing 2 changed files with 4 additions and 2 deletions.
1 change: 0 additions & 1 deletion setup.py
Original file line number Diff line number Diff line change
Expand Up @@ -167,7 +167,6 @@ def _setup_entry_points() -> Dict:
entry_points = {
"console_scripts": [
# sparsification
"sparseml.benchmark=sparseml.benchmark.info:_main",
"sparseml.framework=sparseml.framework.info:_main",
"sparseml.sparsification=sparseml.sparsification.info:_main",
]
Expand Down
5 changes: 4 additions & 1 deletion src/sparseml/benchmark/info.py
Original file line number Diff line number Diff line change
Expand Up @@ -104,7 +104,7 @@
BenchmarkResult,
)
from sparseml.framework.info import FrameworkInferenceProviderInfo, FrameworkInfo
from sparseml.utils import clean_path, create_parent_dirs
from sparseml.utils import clean_path, create_parent_dirs, deprecation_warning
from sparseml.utils.helpers import convert_to_bool


Expand Down Expand Up @@ -544,6 +544,9 @@ def _parse_args():


def _main():
deprecation_warning(
message=f"{__file__} is scheduled for deprecation in a future version",
)
args = _parse_args()
save_benchmark_results(
model=args.model,
Expand Down

0 comments on commit 137abc0

Please sign in to comment.