[CI] Fix smoke tests

ghstack-source-id: d529689fe2b08bed90f55cb2edd0592571619d85
Pull Request resolved: #1147

test/smoke_test.py

@@ -2,10 +2,6 @@
 #
 # This source code is licensed under the MIT license found in the
 # LICENSE file in the root directory of this source tree.
-import argparse
-import sys
-
-import pytest
 
 
 def test_imports():
@@ -22,5 +18,4 @@ def test_imports():
 
 
 if __name__ == "__main__":
-    args, unknown = argparse.ArgumentParser().parse_known_args()
-    pytest.main([__file__, "--capture", "no", "--exitfirst"] + unknown)
+    test_imports()