pytorch
diff --git a/‎setup.py
-1 b/‎setup.py
-1
diff --git a/‎torchelastic/tsm/__init__.py
-7 b/‎torchelastic/tsm/__init__.py
-7
diff --git a/‎torchelastic/tsm/driver/test/__init__.py
-7 b/‎torchelastic/tsm/driver/test/__init__.py
-7
@@ -45,7 +45,6 @@ def get_version():
         install_requires=reqs.strip().split("\n"),
         include_package_data=True,
         packages=find_packages(exclude=("*.test", "aws*", "*.fb")),
-        test_suite="torchelastic.tsm.test.suites.unittests",
         # PyPI package information.
         classifiers=[
             "Development Status :: 4 - Beta",