Browse Source

fix tests

Alex Cheema 1 year ago
parent
commit
b9a2c0f730
2 changed files with 7 additions and 6 deletions
  1. 1 1
      .github/workflows/test.yml
  2. 6 5
      exo/inference/test_inference_engine.py

+ 1 - 1
.github/workflows/test.yml

@@ -51,7 +51,7 @@ jobs:
         kill $PID1 $PID2
 
         # Check outputs
-        if grep -q "Discovered new peer" output1.log && grep -q "Device discovered" output2.log; then
+        if grep -q "Connected to peer" output1.log && grep -q "Connected to peer" output2.log; then
           echo "Test passed: Both instances discovered each other"
           exit 0
         else

+ 6 - 5
exo/inference/test_inference_engine.py

@@ -30,8 +30,9 @@ asyncio.run(test_inference_engine(
     "mlx-community/Meta-Llama-3-8B-Instruct-4bit",
 ))
 
-asyncio.run(test_inference_engine(
-    TinygradDynamicShardInferenceEngine(),
-    TinygradDynamicShardInferenceEngine(),
-    "llama3-8b-sfr",
-))
+# TODO: Waiting on https://github.com/tinygrad/tinygrad/issues/5549
+# asyncio.run(test_inference_engine(
+#     TinygradDynamicShardInferenceEngine(),
+#     TinygradDynamicShardInferenceEngine(),
+#     "llama3-8b-sfr",
+# ))