<img width="1100" height="151" alt="Image" src="https://github.com/user-attachments/assets/1fac2bfc-99ed-44a7-9c0e-6b47094945b0" /> if you run full-shot.py by using "--novel_class", and not_full_shot parameter is false. The novel_class branch never gets exectuded. <img width="1079" height="508" alt="Image" src="https://github.com/user-attachments/assets/d2ba0338-7d22-4f8b-9c07-9fc55fe17f31" />