Add pytest

inference-labs-inc · Jan 17, 2024 · 3b9b1df · 3b9b1df
1 parent 7b3ba26
commit 3b9b1df
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/Dockerfile b/Dockerfile
@@ -19,7 +19,8 @@ RUN pip install \
     'hummingbird-ml' \
     'sk2torch' \
     'web3' \
-    'py-solc-x'
+    'py-solc-x' \
+    'pytest'
 
 RUN git clone https://github.com/zkonduit/ezkl.git && \
     mv ezkl/examples/notebooks/* /home/jovyan/work && \

diff --git a/README.md b/README.md
@@ -53,6 +53,7 @@ These packages are preinstalled in the container along with packages from the [j
 | `sk2torch` | [GitHub](https://github.com/InferenceLabs/sk2torch) |
 | `web3` | [GitHub](https://github.com/ethereum/web3.py) |
 | `py-solc-x` | [GitHub](https://github.com/iamdefinitelyahuman/py-solc-x) |
+| `pytest` | [GitHub](https://github.com/pytest-dev/pytest) |
 
 [`EZKL`]: https://github.com/zkonduit/ezkl "EZKL"
 [jupyter/pytorch-notebook]: https://github.com/jupyter/docker-stacks/tree/main/images/pytorch-notebook "PyTorch Notebook"