Merge pull request #158 from mlcommons/hta-bug-fix

Add missing dependency to HTA
mlcommons · Oct 8, 2024 · f256593 · f256593
2 parents 4a4fd09 + 269a2af
commit f256593
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/pyproject.toml b/pyproject.toml
@@ -11,7 +11,13 @@ license = {file = "LICENSE.md"}
 authors = [
     {name = "MLCommons", email = "[email protected]"},
 ]
-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"