From 8423fc9c9541d817c5ea20da1c8d76167b6c024e Mon Sep 17 00:00:00 2001 From: bcoutinho Date: Wed, 4 Jun 2025 22:58:16 -0700 Subject: [PATCH] add toml and readme changes Summary: As above X-link: https://github.com/pytorch-labs/chakra_replay/pull/3 Reviewed By: shengfukevin Differential Revision: D75995926 Pulled By: briancoutinho --- et_replay/README.md | 5 ++--- et_replay/__init__.py | 3 --- et_replay/pyproject.toml | 2 +- et_replay/tests/test_execution_trace.py | 2 +- 4 files changed, 4 insertions(+), 8 deletions(-) diff --git a/et_replay/README.md b/et_replay/README.md index b44cbda7..ce9b4a9a 100644 --- a/et_replay/README.md +++ b/et_replay/README.md @@ -5,12 +5,11 @@ To install `et_replay`, use the following commands: ```bash -$ git clone --recurse-submodules git@github.com:facebookresearch/param.git +$ git clone https://github.com/pytorch-labs/chakra_replay/ $ conda create -n et_replay python=3.10 $ conda activate et_replay -$ cd param +$ cd chakra_replay $ pip3 install -r requirements.txt -$ cd et_replay $ pip3 install . ``` diff --git a/et_replay/__init__.py b/et_replay/__init__.py index e9864914..e69de29b 100644 --- a/et_replay/__init__.py +++ b/et_replay/__init__.py @@ -1,3 +0,0 @@ -from et_replay.execution_trace import ExecutionTrace - -__all__ = ["ExecutionTrace"] diff --git a/et_replay/pyproject.toml b/et_replay/pyproject.toml index 24838414..dad804b2 100644 --- a/et_replay/pyproject.toml +++ b/et_replay/pyproject.toml @@ -12,7 +12,7 @@ dependencies = [ ] [tool.setuptools.package-dir] -"et_replay" = "." +"et_replay" = "et_replay" [project.scripts] comm_replay = "et_replay.tools.comm_replay:main" diff --git a/et_replay/tests/test_execution_trace.py b/et_replay/tests/test_execution_trace.py index 43a4589f..fea04304 100644 --- a/et_replay/tests/test_execution_trace.py +++ b/et_replay/tests/test_execution_trace.py @@ -3,7 +3,7 @@ import os import unittest -from et_replay import ExecutionTrace +from et_replay.execution_trace import ExecutionTrace from et_replay.tools.validate_trace import TraceValidator CURR_DIR = os.path.dirname(os.path.realpath(__file__))