ray/ci/env/check_minimal_install.py
Kai Fricke 65d9a410f7
[ci] Clean up ci/ directory (refactor ci/travis) (#23866)
Clean up the ci/ directory. This means getting rid of the travis/ path completely and moving the files into sensible subdirectories.

Details:

- Moves everything under ci/travis into subdirectories, e.g. ci/build, ci/lint, etc.
- Minor adjustments to some scripts (variable renames)
- Removes the outdated (unused) asan tests
2022-04-13 18:11:30 +01:00

47 lines
1.3 KiB
Python

"""
This script ensures that some dependencies are _not_ installed in the
current python environment.
This is to ensure that tests with minimal dependencies are not tainted
by too many installed packages.
"""
from typing import List
# These are taken from `setup.py` for ray[default]
DEFAULT_BLACKLIST = [
"aiohttp",
"aiohttp_cors",
"colorful",
"py-spy",
"gpustat",
"opencensus",
"prometheus_client",
"smart_open",
]
def assert_packages_not_installed(blacklist: List[str]):
try:
from pip._internal.operations import freeze
except ImportError: # pip < 10.0
from pip.operations import freeze
installed_packages = [p.split("==")[0].split(" @ ")[0] for p in freeze.freeze()]
assert not any(p in installed_packages for p in blacklist), (
f"Found blacklisted packages in installed python packages: "
f"{[p for p in blacklist if p in installed_packages]}. "
f"Minimal dependency tests could be tainted by this. "
f"Check the install logs and primary dependencies if any of these "
f"packages were installed as part of another install step."
)
print(
f"Confirmed that blacklisted packages are not installed in "
f"current Python environment: {blacklist}"
)
if __name__ == "__main__":
assert_packages_not_installed(DEFAULT_BLACKLIST)