fix: install openenv-core extra so forgeenv.env imports
Browse files
scripts/jobs/train_repair_agent.py
CHANGED
|
@@ -69,11 +69,12 @@ _sh([
|
|
| 69 |
sys.path.insert(0, str(src_dir))
|
| 70 |
|
| 71 |
step("1. pip install + verify GPU")
|
| 72 |
-
_sh([sys.executable, "-m", "pip", "install", "-e",
|
| 73 |
_sh([
|
| 74 |
sys.executable, "-m", "pip", "install",
|
| 75 |
"trl", "peft", "accelerate", "datasets", "bitsandbytes",
|
| 76 |
"matplotlib", "pyyaml", "nltk", "scikit-learn",
|
|
|
|
| 77 |
])
|
| 78 |
try:
|
| 79 |
_sh([sys.executable, "-m", "pip", "install", "unsloth"])
|
|
|
|
| 69 |
sys.path.insert(0, str(src_dir))
|
| 70 |
|
| 71 |
step("1. pip install + verify GPU")
|
| 72 |
+
_sh([sys.executable, "-m", "pip", "install", "-e", f"{src_dir}[openenv]"])
|
| 73 |
_sh([
|
| 74 |
sys.executable, "-m", "pip", "install",
|
| 75 |
"trl", "peft", "accelerate", "datasets", "bitsandbytes",
|
| 76 |
"matplotlib", "pyyaml", "nltk", "scikit-learn",
|
| 77 |
+
"openenv-core>=0.2.0",
|
| 78 |
])
|
| 79 |
try:
|
| 80 |
_sh([sys.executable, "-m", "pip", "install", "unsloth"])
|