Eric Botti
commited on
Commit
·
f12d2e1
1
Parent(s):
cc401d9
removed print
Browse files- src/data_collection.py +2 -0
src/data_collection.py
CHANGED
|
@@ -22,6 +22,8 @@ Model = NewType("Model", BaseModel)
|
|
| 22 |
def save(log_object: Model):
|
| 23 |
collection = get_collection(log_object)
|
| 24 |
|
|
|
|
|
|
|
| 25 |
if DATA_COLLECTION_MODE.upper() == "JSONL":
|
| 26 |
data_dir = pathlib.Path(__file__).parent.parent / "data"
|
| 27 |
log_file = os.path.join(data_dir, f"{collection}.jsonl")
|
|
|
|
| 22 |
def save(log_object: Model):
|
| 23 |
collection = get_collection(log_object)
|
| 24 |
|
| 25 |
+
print(DATA_COLLECTION_MODE)
|
| 26 |
+
|
| 27 |
if DATA_COLLECTION_MODE.upper() == "JSONL":
|
| 28 |
data_dir = pathlib.Path(__file__).parent.parent / "data"
|
| 29 |
log_file = os.path.join(data_dir, f"{collection}.jsonl")
|