awslabs · chhzh123 · May 19, 2023 · May 21, 2023 · May 21, 2023 · May 21, 2023
diff --git a/ci/install_test_pkgs.sh b/ci/install_test_pkgs.sh
@@ -5,3 +5,4 @@
 python3 -m pip install black==22.10.0
 python3 -m pip install transformers==4.25.1 --no-deps
 python3 -m pip install pylint==2.14.0 astroid==2.11.6 mock==4.0.3
+python3 -m pip install z3-solver tabulate
diff --git a/setup.py b/setup.py
@@ -128,6 +128,8 @@ def setup():
         long_description_content_type="text/markdown",
         setup_requires=[],
         install_requires=[
+            "z3-solver",
+            "tabulate",
             "packaging",
             "psutil",
         ],

diff --git a/slapo/sharding/__init__.py b/slapo/sharding/__init__.py
@@ -10,3 +10,4 @@
     scatter_forward_output,
     reduce_forward_output,
 )
+from .solver import Solver