-
Notifications
You must be signed in to change notification settings - Fork 93
/
any_env2.py
145 lines (123 loc) · 6.34 KB
/
any_env2.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
"""Modify dataset with arbitrary env"""
from h2oaicore.data import CustomData
import functools
def wrap_create(pyversion="3.11", install_h2oaicore=False, install_datatable=True, modules_needed_by_name=[],
cache_env=False, file=None, id=None,
**kwargs_wrapper):
""" Decorate a function to create_data in popen in isolated env
"""
def actual_decorator(func):
@functools.wraps(func)
def wrapper(*args, **kwargs):
return create_data_popen(func, *args, **kwargs, pyversion=pyversion, install_h2oaicore=install_h2oaicore,
install_datatable=install_datatable, modules_needed_by_name=modules_needed_by_name,
cache_env=cache_env, file=file, id=id,
**kwargs_wrapper)
return wrapper
return actual_decorator
def create_data_popen(func, *args, pyversion="3.11", install_h2oaicore=False, install_datatable=True,
modules_needed_by_name=[], cache_env=False, file=None, id=None,
X=None, **kwargs):
""" Run recipe in popen in isolated env
"""
print(
"Recipe %s running pyversion=%s install_h2oaicore=%s install_datatable=%s modules_needed_by_name=%s cache_env=%s id=%s" % (
file, pyversion, install_h2oaicore, install_datatable,
modules_needed_by_name, cache_env, id))
import os
from h2oaicore.data import DataContribLoader
env_dir_orig = DataContribLoader()._env_dir
base_orig = os.path.basename(file).replace(".py", "")
if id is None:
id = base_orig
env_path = os.path.abspath(os.path.join(env_dir_orig, "recipe_env_%s" % id))
use_cache = os.path.isdir(env_path) and cache_env
if use_cache:
print("Using cache at %s for recipe %s" % (env_path, file))
os.makedirs(env_path, exist_ok=True)
X_file = os.path.abspath(os.path.join(env_path, "X.jay"))
Y_file = os.path.abspath(os.path.join(env_path, "Y.jay"))
if X is not None:
X.to_jay(X_file)
print("X.names: %s" % (list(X.names)))
python_script_file = os.path.abspath(os.path.join(env_path, "script.py"))
with open(os.path.abspath(__file__), "rt") as src:
with open(python_script_file, "wt") as dst:
for line in src.readlines():
if line.startswith("from h2oaicore.data import CustomData"):
continue
if line.endswith("CustomData):\n"):
line = line.replace("CustomData", "object")
assert line.startswith("class ")
class_name = line[6:-10]
if line.startswith(" @wrap_create"):
continue
dst.write(line)
script_name = os.path.abspath(os.path.join(env_path, "script.sh"))
import os
with open(script_name, "wt") as f:
print("set -o pipefail", file=f)
print("set -ex", file=f)
print("unset PYTHONPATH", file=f)
print("unset PYTHONUSERBASE", file=f)
print("mkdir -p %s" % env_path, file=f)
if not use_cache:
print("virtualenv -p python%s %s" % (pyversion, env_path), file=f)
print("source %s/bin/activate" % env_path, file=f)
if not use_cache:
print("python -m pip install --upgrade pip", file=f)
print("%s/bin/python -m pip debug --verbose" % (env_path), file=f)
print("%s/bin/python -c \'import platform ; print(platform.architecture())\'" % (env_path), file=f)
dai_home = os.environ.get('DRIVERLESS_AI_HOME', os.getcwd())
template_dir = os.environ.get("H2OAI_SCORER_TEMPLATE_DIR", os.path.join(dai_home, 'h2oai_scorer'))
if install_h2oaicore:
print("pip install %s" % os.path.join(template_dir, 'scoring-pipeline', 'license-*'), file=f)
print("pip install %s" % os.path.join(template_dir, 'scoring-pipeline', 'h2oaicore-*'), file=f)
if install_datatable:
print("pip install %s" % os.path.join(template_dir, 'scoring-pipeline', 'datatable-*'), file=f)
for pkg in modules_needed_by_name:
print("%s/bin/pip install %s --ignore-installed" % (env_path, pkg), file=f)
print("cd %s" % os.path.dirname(python_script_file), file=f)
script_module_name = os.path.basename(python_script_file.replace(".py", ""))
print(
"%s/bin/python -c \'from %s import %s ; Y = %s.%s(\"%s\") ; import datatable as dt ; Y.to_jay(\"%s\")\'" % (
env_path, script_module_name, class_name, class_name, func.__name__, X_file, Y_file), file=f)
import stat
os.chmod(script_name, stat.S_IRWXU)
syntax = [script_name]
from h2oaicore.systemutils import FunnelPopen
with FunnelPopen(syntax, shell=True) as fp:
fp.launch().read()
import datatable as dt
Y = dt.fread(Y_file)
from h2oaicore.systemutils import remove
remove(X_file)
remove(Y_file)
if not cache_env:
remove(env_path)
return Y
class FreshEnvData(CustomData):
@staticmethod
# Specify the python package dependencies. Will be installed in order of list
# NOTE: Keep @wrap_create on a single line
# NOTE: If want to share cache across recipes, can set cache_env=True and set id=<some unique identifier, like myrecipe12345>
# Below caches the env into "id" folder
# @wrap_create(pyversion="3.11", install_h2oaicore=False, install_datatable=True, modules_needed_by_name=["pandas==1.5.3"], cache_env=True, file=__file__, id="myrecipe12345")
# Below does not cache the env
@wrap_create(pyversion="3.11", install_h2oaicore=False, install_datatable=True, modules_needed_by_name=["pandas==1.5.3"], file=__file__)
def create_data(X=None):
import os
import datatable as dt
if X is not None and os.path.isfile(X):
X = dt.fread(X)
else:
X = None
my_path = os.path.dirname(__file__)
import pandas as pd
assert pd.__version__ == "1.1.5", "actual: %s" % pd.__version__
url = "http://data.un.org/_Docs/SYB/CSV/SYB63_226_202009_Net%20Disbursements%20from%20Official%20ODA%20to%20Recipients.csv"
import urllib.request
new_file = os.path.join(my_path, "user_file.csv")
urllib.request.urlretrieve(url, new_file)
import datatable as dt
return dt.fread(new_file)