Eric Botti commited on
Commit
daaf6ff
·
2 Parent(s): aeb8841 5aea999

Merge branch 'main' into space

Browse files
Files changed (1) hide show
  1. src/data_collection.py +5 -4
src/data_collection.py CHANGED
@@ -9,7 +9,7 @@ import player
9
 
10
  from pydantic import BaseModel
11
 
12
-
13
 
14
  Model = NewType("Model", BaseModel)
15
 
@@ -18,10 +18,11 @@ data_dir = pathlib.Path(__file__).parent.parent / "data"
18
 
19
 
20
  def save(log_object: Model):
21
- log_file = get_log_file(log_object)
 
22
 
23
- with open(log_file, "a+") as f:
24
- f.write(log_object.model_dump_json() + "\n")
25
 
26
 
27
  def get_log_file(log_object: Model) -> str:
 
9
 
10
  from pydantic import BaseModel
11
 
12
+ COLLECT_DATA = os.environ.get("COLLECT_DATA", "true").upper() == "TRUE"
13
 
14
  Model = NewType("Model", BaseModel)
15
 
 
18
 
19
 
20
  def save(log_object: Model):
21
+ if COLLECT_DATA:
22
+ log_file = get_log_file(log_object)
23
 
24
+ with open(log_file, "a+") as f:
25
+ f.write(log_object.model_dump_json() + "\n")
26
 
27
 
28
  def get_log_file(log_object: Model) -> str: