Only run vaex tests if vaex is installed

Author: marthacryan

tests/test_optional/test_px/test_px_input.py

@@ -14,7 +14,7 @@
 import warnings
 
 
-if optional_imports.get_module("vaex") is None and sys.version_info == (3, 9):
+if optional_imports.get_module("vaex"):
     TEST_LIBS = ["polars"]
 else:
     TEST_LIBS = ["vaex", "polars"]