Skip to content

Refactor: Move traceback imports to top of files #5

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
GilbertKrantz merged 2 commits into from
Oct 24, 2025
Merged

Refactor: Move traceback imports to top of files #5

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
2 commits
Select commit Hold shift + click to select a range
bae3f67
Initial plan
Copilot Oct 23, 2025
28ce8ce
Refactor: Move traceback imports to top of files
Copilot Oct 23, 2025
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
3 changes: 1 addition & 2 deletions gradio-inference.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -5,6 +5,7 @@
import os
import numpy as np
import cv2
import traceback

import torch
import torch.nn as nn
Expand Down Expand Up @@ -219,7 +220,6 @@ def predict_image(image: np.ndarray, model_type: str) -> tuple[dict, np.ndarray]

except Exception as e:
logging.error(f"Error generating heatmap: {e}")
import traceback
traceback.print_exc()
# Fallback: just do prediction without heatmap
with torch.no_grad():
Expand All @@ -237,7 +237,6 @@ def predict_image(image: np.ndarray, model_type: str) -> tuple[dict, np.ndarray]

except Exception as e:
logging.error(f"Error during prediction: {e}")
import traceback
traceback.print_exc()
return {cls: 0.0 for cls in CLASSES}, None

Expand Down
5 changes: 1 addition & 4 deletions utils/Trainer.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -6,6 +6,7 @@
from torch.utils.data import DataLoader
from tqdm import tqdm
import gc
import traceback

from utils.Evaluator import ClassificationEvaluator
from utils.Callback import EarlyStopping
Expand Down Expand Up @@ -272,15 +273,11 @@ def model_train(
return results
except Exception as viz_error:
print(f"Error in visualization: {viz_error}")
import traceback

traceback.print_exc()
return {"accuracy": None}

except Exception as e:
print(f"Error occurred when training {model_name}: {e}")
import traceback

traceback.print_exc()
return {"accuracy": None}
finally:
Expand Down