diff --git a/pyproject.toml b/pyproject.toml
index 0833e896..8fa00717 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -11,7 +11,13 @@ license = {file = "LICENSE.md"}
 authors = [
     {name = "MLCommons", email = "chakra@mlcommons.org"},
 ]
-dependencies = ["protobuf==5.*", "graphviz", "networkx", "pydot"]
+dependencies = [
+    "protobuf==5.*",
+    "graphviz",
+    "networkx",
+    "pydot",
+    "HolisticTraceAnalysis @ git+https://github.com/facebookresearch/HolisticTraceAnalysis.git@d731cc2e2249976c97129d409a83bd53d93051f6"
+]
 
 [project.urls]
 Homepage = "https://github.com/mlcommons/chakra"