Function refactoring

facebookresearch · May 14, 2024 · cb11256 · cb11256
1 parent 7feeea8
commit cb11256
Show file tree

Hide file tree

Showing 28 changed files with 3,039 additions and 5,418 deletions.
diff --git a/.github/workflows/python_lint.yml b/.github/workflows/python_lint.yml
@@ -1,6 +1,6 @@
 name: Python Linting
 
-on: [push, pull_request]
+on: pull_request
 
 jobs:
   lint-and-format:
@@ -18,4 +18,7 @@ jobs:
       run: |
         pip install black
     - name: Run Black
-      run: black . --check
+      run: black . --check --exclude et_replay
+
+    - name: Run Ruff format
+      run: ruff format et_replay
diff --git a/et_replay/.gitignore b/et_replay/.gitignore
@@ -1,2 +1,3 @@
 build/
 et_replay.egg-info/
+__pycache__/
diff --git a/et_replay/lib/__init__.py b/et_replay/lib/__init__.py
diff --git a/et_replay/lib/chakra_trace_parser.py b/et_replay/lib/chakra_trace_parser.py
diff --git a/et_replay/lib/comm/__init__.py b/et_replay/lib/comm/__init__.py
@@ -0,0 +1,3 @@
+from .comm_op_args import CommOpArgs
+from .comm_tensor_allocator import CommTensorAllocator
+from .comm_trace_reader import CommTraceReader
diff --git a/et_replay/lib/comm/backend/__init__.py b/et_replay/lib/comm/backend/__init__.py
@@ -0,0 +1,3 @@
+from .base_backend import BaseBackend
+from .mock import MockBackend
+from .pytorch_dist import PyTorchDistBackend