Eric Botti commited on
Commit
752e673
·
1 Parent(s): f7ce19f

data dir mapped from current filepath

Browse files
Files changed (1) hide show
  1. 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 = os.path.join(os.pardir, "data")
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):