diff --git a/gpt_engineer/collect.py b/gpt_engineer/collect.py index 118e556..865416a 100644 --- a/gpt_engineer/collect.py +++ b/gpt_engineer/collect.py @@ -23,8 +23,8 @@ def send_learning(learning: Learning): def collect_learnings(model: str, temperature: float, steps: List[Step], dbs: DBs): - if os.environ.get("COLLECT_LEARNINGS_OPT_OUT") in ["true", "1"]: - print("COLLECT_LEARNINGS_OPT_OUT is set to true, not collecting learning") + if os.environ.get("COLLECT_LEARNINGS_OPT_IN") in ["false", "1"]: + print("COLLECT_LEARNINGS_OPT_IN is set to false, not collecting learning") return learnings = extract_learning( diff --git a/tests/test_collect.py b/tests/test_collect.py index 761e002..e75cbc1 100644 --- a/tests/test_collect.py +++ b/tests/test_collect.py @@ -13,7 +13,7 @@ from gpt_engineer.steps import gen_code def test_collect_learnings(monkeypatch): - monkeypatch.setattr(os, "environ", {"COLLECT_LEARNINGS_OPT_OUT": "false"}) + monkeypatch.setattr(os, "environ", {"COLLECT_LEARNINGS_OPT_IN": "true"}) monkeypatch.setattr(rudder_analytics, "track", MagicMock()) model = "test_model"