mirror of
https://github.com/vale981/ray
synced 2025-03-06 18:41:40 -05:00
49 lines
1.4 KiB
Python
49 lines
1.4 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",
|
||
|
|
"aioredis",
|
||
|
|
"colorful",
|
||
|
|
"py-spy",
|
||
|
|
"jsonschema",
|
||
|
|
# "requests",
|
||
|
|
"gpustat",
|
||
|
|
"opencensus",
|
||
|
|
"prometheus_client",
|
||
|
|
]
|
||
|
|
|
||
|
|
|
||
|
|
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)
|