Rename *2chakra_converter to *_converter

mlcommons · May 9, 2024 · 95afc4f · 95afc4f
1 parent 705e00c
commit 95afc4f
Show file tree

Hide file tree

Showing 4 changed files with 7 additions and 9 deletions.
diff --git a/pyproject.toml b/pyproject.toml
@@ -44,7 +44,7 @@ target-version = "py39"
 line-length = 120
 exclude = [
     "src/third_party/protolib.py",
-    "src/converter/text2chakra_converter.py",
+    "src/converter/text_converter.py",
 ]
 
 [tool.ruff.lint]

diff --git a/src/converter/converter.py b/src/converter/converter.py
@@ -6,8 +6,8 @@
 import traceback
 from logging import FileHandler
 
-from .pytorch2chakra_converter import PyTorch2ChakraConverter
-from .text2chakra_converter import Text2ChakraConverter
+from .pytorch_converter import PyTorchConverter
+from .text_converter import TextConverter
 
 
 def get_logger(log_filename: str) -> logging.Logger:
@@ -52,12 +52,10 @@ def main() -> None:
 
     try:
         if args.input_type == "Text":
-            converter = Text2ChakraConverter(
-                args.input_filename, args.output_filename, args.num_npus, args.num_passes, logger
-            )
+            converter = TextConverter(args.input_filename, args.output_filename, args.num_npus, args.num_passes, logger)
             converter.convert()
         elif args.input_type == "PyTorch":
-            converter = PyTorch2ChakraConverter(args.input_filename, args.output_filename, logger)
+            converter = PyTorchConverter(args.input_filename, args.output_filename, logger)
             converter.convert()
         else:
             logger.error(f"{args.input_type} unsupported")

diff --git a/src/converter/pytorch2chakra_converter.py → src/converter/pytorch_converter.py b/src/converter/pytorch2chakra_converter.py → src/converter/pytorch_converter.py
@@ -27,7 +27,7 @@
 from .pytorch_node import PyTorchNode, PyTorchNodeType
 
 
-class PyTorch2ChakraConverter:
+class PyTorchConverter:
     """
     Converter class for transforming PyTorch execution traces into Chakra format.
 

diff --git a/src/converter/text2chakra_converter.py → src/converter/text_converter.py b/src/converter/text2chakra_converter.py → src/converter/text_converter.py
@@ -50,7 +50,7 @@ def __init__(self, line: str) -> None:
             raise ValueError(f'Cannot parse the following layer -- "{line}"')
 
 
-class Text2ChakraConverter:
+class TextConverter:
     def __init__(
         self, input_filename: str, output_filename: str, num_npus: int, num_passes: int, logger: logging.Logger
     ) -> None: