runtime error

Exit code: 1. Reason: Traceback (most recent call last): File "/home/user/app/app.py", line 5, in <module> class DDPMCB(TrainCB): NameError: name 'TrainCB' is not defined

Container logs: