diff --git a/packaging/post_build_script.sh b/packaging/post_build_script.sh index e036fcbcfe..de8ee956f2 100644 --- a/packaging/post_build_script.sh +++ b/packaging/post_build_script.sh @@ -30,6 +30,4 @@ popd MANYWHEEL_NAME=$(ls dist/) # Try to install the new wheel pip install "dist/${MANYWHEEL_NAME}" -# and validating it by running the unit tests. Some tests are failing here and -# there, so let's add more of them later -pytest -v test/test_ops.py +python -c "import torchao" diff --git a/packaging/pre_build_script.sh b/packaging/pre_build_script.sh index 366752f7ed..4bbd09694a 100644 --- a/packaging/pre_build_script.sh +++ b/packaging/pre_build_script.sh @@ -10,5 +10,3 @@ set -eux echo "This script is run before building torchao binaries" pip install setuptools wheel twine auditwheel -pip install -r requirements.txt -pip install -r dev-requirements.txt diff --git a/packaging/smoke_test.py b/packaging/smoke_test.py index 2c2c873512..8ffa39ffde 100644 --- a/packaging/smoke_test.py +++ b/packaging/smoke_test.py @@ -6,14 +6,14 @@ # LICENSE file in the root directory of this source tree. import subprocess -import torchao.ops +import torchao def main(): """ Run torchao binary smoke tests like importing and performing simple ops """ - print(dir(torchao.ops)) + print(dir(torchao)) if __name__ == "__main__":