ethix commited on
Commit
46bfcc6
·
1 Parent(s): 15b57f6

fix: enhance save_image function with type hint and debug prints

Browse files
Files changed (1) hide show
  1. debug.py +5 -3
debug.py CHANGED
@@ -28,8 +28,10 @@ scheduler = CommitScheduler(
28
 
29
  )
30
 
31
- def save_image(image_id, image_path) -> None:
32
  os.makedirs(IMAGE_DIR, exist_ok=True)
 
 
33
  save_image_path = os.path.join(IMAGE_DIR, f"{image_id}.png")
34
  with scheduler.lock:
35
  try:
@@ -55,8 +57,8 @@ def log_params(
55
  print("Type after: " + str(type(after_image)))
56
  save_image(before_id, before_image)
57
  save_image(after_id, after_image)
58
- print("Before path: " + before_path)
59
- print("After path: " + after_path)
60
 
61
  is_new = not os.path.exists(LOG_FILE)
62
  with open(LOG_FILE, "a", newline='') as f:
 
28
 
29
  )
30
 
31
+ def save_image(image_id, image_path: Image.Image) -> None:
32
  os.makedirs(IMAGE_DIR, exist_ok=True)
33
+ print("Image ID: " + image_id)
34
+ print("Image ID Type: " + str(type(image_id)))
35
  save_image_path = os.path.join(IMAGE_DIR, f"{image_id}.png")
36
  with scheduler.lock:
37
  try:
 
57
  print("Type after: " + str(type(after_image)))
58
  save_image(before_id, before_image)
59
  save_image(after_id, after_image)
60
+ #print("Before path: " + before_path)
61
+ #print("After path: " + after_path)
62
 
63
  is_new = not os.path.exists(LOG_FILE)
64
  with open(LOG_FILE, "a", newline='') as f: