Eric Botti
commited on
Commit
·
752e673
1
Parent(s):
f7ce19f
data dir mapped from current filepath
Browse files- src/data_collection.py +4 -1
src/data_collection.py
CHANGED
@@ -1,4 +1,5 @@
|
|
1 |
import os
|
|
|
2 |
from typing import NewType
|
3 |
|
4 |
import pydantic
|
@@ -8,10 +9,12 @@ import player
|
|
8 |
|
9 |
from pydantic import BaseModel
|
10 |
|
|
|
|
|
11 |
Model = NewType("Model", BaseModel)
|
12 |
|
13 |
|
14 |
-
data_dir =
|
15 |
|
16 |
|
17 |
def save(log_object: Model):
|
|
|
1 |
import os
|
2 |
+
import pathlib
|
3 |
from typing import NewType
|
4 |
|
5 |
import pydantic
|
|
|
9 |
|
10 |
from pydantic import BaseModel
|
11 |
|
12 |
+
|
13 |
+
|
14 |
Model = NewType("Model", BaseModel)
|
15 |
|
16 |
|
17 |
+
data_dir = pathlib.Path(__file__).parent.parent / "data"
|
18 |
|
19 |
|
20 |
def save(log_object: Model):
|