cjber commited on
Commit
3dfa711
·
1 Parent(s): 49a6658

fix: add more paths to regenerate

Browse files
Files changed (1) hide show
  1. planning_ai/common/utils.py +8 -2
planning_ai/common/utils.py CHANGED
@@ -1,3 +1,4 @@
 
1
  from pathlib import Path
2
 
3
  import polars as pl
@@ -30,20 +31,25 @@ class Paths:
30
  STAGING = DATA / "staging"
31
  OUT = DATA / "out"
32
 
 
 
33
  SUMMARY = OUT / "summary"
34
- SUMMARIES = OUT / "summaries"
35
 
36
  PROMPTS = Path("planning_ai/chains/prompts")
37
 
38
  @classmethod
39
  def ensure_directories_exist(cls):
 
 
40
  for path in [
41
  cls.DATA,
42
  cls.RAW,
43
  cls.STAGING,
44
  cls.OUT,
45
  cls.SUMMARY,
46
- cls.SUMMARIES,
 
47
  ]:
48
  path.mkdir(parents=True, exist_ok=True)
49
 
 
1
+ import shutil
2
  from pathlib import Path
3
 
4
  import polars as pl
 
31
  STAGING = DATA / "staging"
32
  OUT = DATA / "out"
33
 
34
+ PDFS_AZURE = STAGING / "pdfs_azure"
35
+
36
  SUMMARY = OUT / "summary"
37
+ FIGS = SUMMARY / "figs"
38
 
39
  PROMPTS = Path("planning_ai/chains/prompts")
40
 
41
  @classmethod
42
  def ensure_directories_exist(cls):
43
+ # for path in [cls.STAGING, cls.OUT]:
44
+ # shutil.rmtree(path, ignore_errors=True)
45
  for path in [
46
  cls.DATA,
47
  cls.RAW,
48
  cls.STAGING,
49
  cls.OUT,
50
  cls.SUMMARY,
51
+ cls.FIGS,
52
+ cls.PDFS_AZURE,
53
  ]:
54
  path.mkdir(parents=True, exist_ok=True)
55