tritonparse 0.2.4.dev20251006071528__tar.gz → 0.2.4.dev20251007071533__tar.gz

{tritonparse-0.2.4.dev20251006071528/tritonparse.egg-info → tritonparse-0.2.4.dev20251007071533}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: tritonparse
-Version: 0.2.4.dev20251006071528
+Version: 0.2.4.dev20251007071533
 Summary: TritonParse: A Compiler Tracer, Visualizer, and mini-Reproducer Generator for Triton Kernels
 Author-email: Yueming Hao <yhao@meta.com>
 License-Expression: BSD-3-Clause

{tritonparse-0.2.4.dev20251006071528 → tritonparse-0.2.4.dev20251007071533}/pyproject.toml

@@ -48,4 +48,4 @@ local_scheme = "no-local-version"
 "Homepage" = "https://github.com/meta-pytorch/tritonparse"
 
 [project.scripts]
-tritonparse = "tritonparse.run:main"
+tritonparseoss = "tritonparse.cli:main"

tritonparse-0.2.4.dev20251007071533/run.py

@@ -0,0 +1,8 @@
+#!/usr/bin/env python3
+#  Copyright (c) Meta Platforms, Inc. and affiliates.
+
+from tritonparse.cli import main
+
+if __name__ == "__main__":
+    # Do not add code here, it won't be run. Add them to the function called below.
+    main()  # pragma: no cover

{tritonparse-0.2.4.dev20251006071528 → tritonparse-0.2.4.dev20251007071533}/tritonparse/__main__.py

@@ -1,4 +1,4 @@
-from .run import main
+from .cli import main
 
 
 if __name__ == "__main__":

tritonparse-0.2.4.dev20251006071528/run.py → tritonparse-0.2.4.dev20251007071533/tritonparse/cli.py

@@ -1,9 +1,9 @@
 #!/usr/bin/env python3
 #  Copyright (c) Meta Platforms, Inc. and affiliates.
-
 import argparse
 from importlib.metadata import PackageNotFoundError, version
 
+from .common import is_fbcode
 from .reproducer.cli import _add_reproducer_args
 from .reproducer.orchestrator import reproduce
 from .utils import _add_parse_args, unified_parse
@@ -19,15 +19,18 @@ def _get_package_version() -> str:
 def main():
     pkg_version = _get_package_version()
 
+    # Use different command name for fbcode vs OSS
+    prog_name = "tritonparse" if is_fbcode() else "tritonparseoss"
+
     parser = argparse.ArgumentParser(
-        prog="tritonparse",
+        prog=prog_name,
         description=(
             "TritonParse: parse structured logs and generate minimal reproducers"
         ),
         epilog=(
             "Examples:\n"
-            "  tritonparse parse /path/to/logs --out parsed_output\n"
-            "  tritonparse reproduce /path/to/trace.ndjson --line 1 --out-dir repro_output\n"
+            f"  {prog_name} parse /path/to/logs --out parsed_output\n"
+            f"  {prog_name} reproduce /path/to/trace.ndjson --line 2 --out-dir repro_output\n"
         ),
         formatter_class=argparse.RawDescriptionHelpFormatter,
     )

{tritonparse-0.2.4.dev20251006071528 → tritonparse-0.2.4.dev20251007071533/tritonparse.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: tritonparse
-Version: 0.2.4.dev20251006071528
+Version: 0.2.4.dev20251007071533
 Summary: TritonParse: A Compiler Tracer, Visualizer, and mini-Reproducer Generator for Triton Kernels
 Author-email: Yueming Hao <yhao@meta.com>
 License-Expression: BSD-3-Clause

{tritonparse-0.2.4.dev20251006071528 → tritonparse-0.2.4.dev20251007071533}/tritonparse.egg-info/SOURCES.txt

@@ -36,6 +36,7 @@ tests/example_output/parsed_output_complex/log_file_list.json
 tests/example_output/repro/repro_context_20250816192455.json
 tritonparse/__init__.py
 tritonparse/__main__.py
+tritonparse/cli.py
 tritonparse/common.py
 tritonparse/context_manager.py
 tritonparse/event_diff.py

tritonparse-0.2.4.dev20251007071533/tritonparse.egg-info/entry_points.txt

@@ -0,0 +1,2 @@
+[console_scripts]
+tritonparseoss = tritonparse.cli:main

tritonparse-0.2.4.dev20251006071528/tritonparse.egg-info/entry_points.txt

@@ -1,2 +0,0 @@
-[console_scripts]
-tritonparse = tritonparse.run:main