davanstrien HF Staff commited on
Commit
8c238a2
·
verified ·
1 Parent(s): d285406

update train.py

Browse files
Files changed (1) hide show
  1. train.py +2 -0
train.py CHANGED
@@ -41,6 +41,8 @@ REPO_ID = "harness-race/opencode-r2"
41
 
42
  # ----- trackio (best effort) -----
43
  def setup_trackio():
 
 
44
  try:
45
  import trackio
46
  r = trackio.init(project="opencode-r2", name=os.environ.get("JOB_NAME", "finetune"), private=True)
 
41
 
42
  # ----- trackio (best effort) -----
43
  def setup_trackio():
44
+ if os.environ.get("OPENCODE_TRACKIO", "1") == "0":
45
+ return (lambda **kw: None)
46
  try:
47
  import trackio
48
  r = trackio.init(project="opencode-r2", name=os.environ.get("JOB_NAME", "finetune"), private=True)