import os
from gtd.io import Workspace
# Set location of local data directory from environment variable
env_var = 'TEXTMORPH_DATA'
if env_var not in os.environ:
assert False, env_var + ' environmental variable must be set.'
root = os.environ[env_var]
# define workspace
workspace = Workspace(root)
# config
workspace.add_file('config', 'config.txt')
# Training runs
workspace.add_dir('edit_runs', 'edit_runs')
workspace.add_dir('lm_runs', 'lm_runs')
workspace.add_dir('retriever_runs', 'retriever_runs')
# user IDs
workspace.add_file('user_ids', 'user_ids.json')
# word vectors
workspace.add_dir('word_vectors', 'word_vectors')
# nearest neighbors
workspace.add_dir('nearest_sentences', 'nearest_sentences')
env_var = "HRL_DATA"
if env_var not in os.environ:
assert False, env_var + " environmental variable must be set."
root = os.environ[env_var]
# define workspace
workspace = Workspace(root)
workspace.add_dir("experiments", "experiments")
workspace.add_dir("visualization_rooms", "visualization_rooms")
workspace.add_dir("montezuma_rooms", "visualization_rooms/montezuma")
workspace.add_dir("pitfall_rooms", "visualization_rooms/pitfall")
workspace.add_dir("private_eye_rooms", "visualization_rooms/private_eye")
workspace.add_dir("whitelist", "whitelist")
workspace.add_file("montezuma_whitelist", "whitelist/montezuma.txt")
workspace.add_file("pitfall_whitelist", "whitelist/pitfall.txt")
workspace.add_file("private_eye_whitelist", "whitelist/private_eye.txt")
workspace.add_file("arial", "arial.ttf")
def room_dir(domain):
"""Given a domain, returns the corresponding visualization_rooms subdir.
Args:
domain (str): e.g. MontezumaRevengeNoFrameskip-v4
Returns:
path (str): path to rooms dir
"""