fix script

Signed-off-by: Chen Tong <[email protected]>
onnx · Jan 15, 2025 · a92ccc2 · a92ccc2
1 parent 28584dd
commit a92ccc2
Show file tree

Hide file tree

Showing 3 changed files with 13 additions and 5 deletions.
diff --git a/docs/BuildPyRuntimeLit.md b/docs/BuildPyRuntimeLit.md
@@ -28,8 +28,12 @@ make
 make OMCreatePyRuntimePackage
 
 # Install the package
-pip3 install src/Runtime/python/onnxmlir
+pip3 install -e src/Runtime/python/onnxmlir
+# -e is necessary for current package. Need to add resource description
+# to install the pre-compiled binary
 
 # Run test case
-python3 src/Runtime/python/onnxmlir/test/test_1.py
+cd src/Runtime/python/onnxmlir/tests
+python3 test_1.py
+# Current limitation on where the model is
 ```
diff --git a/src/Runtime/python/onnxmlirdocker.py b/src/Runtime/python/onnxmlirdocker.py
@@ -74,6 +74,7 @@ def __init__(self, model_path, **kwargs):
             self.model_basename.removesuffix(model_suffix),
         )
 
+        print("model: ", self.model_dirname)
         self.container_client = docker.from_env()
         # Logically, the model directory could be mounted as read only.
         # But wrong time error occurred with "r" mode
@@ -91,7 +92,6 @@ def __init__(self, model_path, **kwargs):
                 },
             },
         )
-        print("afterwards tempdir: ", [f for f in os.listdir(self.output_dirname)])
         self.session = self.getSession()
 
     def getSession(self):

diff --git a/utils/build-pyruntime-lit.sh b/utils/build-pyruntime-lit.sh
@@ -9,7 +9,11 @@ make
 make OMCreatePyRuntimePackage
 
 # Install the package
-pip3 install src/Runtime/python/onnxmlir
+pip3 install -e src/Runtime/python/onnxmlir
+# -e is necessary for current package. Need to add resource description
+# to install the pre-compiled binary
 
 # Run test case
-python3 src/Runtime/python/onnxmlir/test/test_1.py
+cd src/Runtime/python/onnxmlir/tests
+python3 test_1.py
+# Current limitation on where the model is