test_utils.py 466 B

123456789101112131415
  1. import os
  2. INITIAL_PEERS = os.environ.get("INITIAL_PEERS")
  3. if not INITIAL_PEERS:
  4. raise RuntimeError("Must specify INITIAL_PEERS environment variable with one or more peer ids")
  5. INITIAL_PEERS = INITIAL_PEERS.split()
  6. MODEL_NAME = os.environ.get("MODEL_NAME")
  7. if not MODEL_NAME:
  8. raise RuntimeError("Must specify MODEL_NAME as an index of a transformer block to be tested")
  9. REF_NAME = os.environ.get("REF_NAME")
  10. ADAPTER_NAME = os.environ.get("ADAPTER_NAME")