Merge branch 'main' into space
Browse files- 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 |
-
|
|
|
22 |
|
23 |
-
|
24 |
-
|
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:
|