Merge pull request mlcommons#22 from mlcommons/pyre

Update pyre configuration and fix errors in et_generator
ajbalogh · Feb 7, 2024 · 13d9961 · 13d9961
2 parents e3a0677 + 5d315b6
commit 13d9961
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/.pyre_configuration b/.pyre_configuration
@@ -1,7 +1,8 @@
 {
   "source_directories": [
     "timeline_visualizer",
-    "et_converter"
+    "et_converter",
+    "utils"
   ],
   "search_path": ["/opt/hostedtoolcache/Python/3.8.18/x64/lib/python3.8/site-packages"]
 }
diff --git a/utils/et_generator/et_generator.py b/utils/et_generator/et_generator.py
@@ -4,6 +4,7 @@
 
 from chakra.third_party.utils.protolib import encodeMessage as encode_message
 from chakra.et_def.et_def_pb2 import (
+    NodeType as ChakraNodeType,
     Node as ChakraNode,
     DoubleList,
     FloatList,
@@ -35,7 +36,7 @@
 
 NODE_ID = 0
 
-def get_node(node_name: str, node_type: int) -> ChakraNode:
+def get_node(node_name: str, node_type: ChakraNodeType) -> ChakraNode:
     global NODE_ID
     node = ChakraNode()
     node.id = NODE_ID