akhiilll commited on
Commit
77cbc5a
·
verified ·
1 Parent(s): a9536eb

fix: install openenv-core extra so forgeenv.env imports

Browse files
Files changed (1) hide show
  1. scripts/jobs/train_repair_agent.py +2 -1
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", str(src_dir)])
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"])