mirror of
https://github.com/twitter/the-algorithm-ml.git
synced 2024-11-16 21:29:24 +01:00
27 lines
500 B
Python
27 lines
500 B
Python
"""This is intended to be run as a module.
|
|
e.g. python -m tml.machines.is_venv
|
|
|
|
Exits with 0 ii running in venv, otherwise 1.
|
|
"""
|
|
|
|
import sys
|
|
import logging
|
|
|
|
|
|
def is_venv():
|
|
# See https://stackoverflow.com/questions/1871549/determine-if-python-is-running-inside-virtualenv
|
|
return sys.base_prefix != sys.prefix
|
|
|
|
|
|
def _main():
|
|
if is_venv():
|
|
logging.info("In venv %s", sys.prefix)
|
|
sys.exit(0)
|
|
else:
|
|
logging.error("Not in venv")
|
|
sys.exit(1)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
_main()
|