Skip to content

fix(engine): resolve f-string syntax error in trainer.py and add MPS support - #25

Open
atharvasardarx wants to merge 1 commit into
ML4SCI:mainfrom
atharvasardarx:feature/improve-training-loop
Open

fix(engine): resolve f-string syntax error in trainer.py and add MPS support#25
atharvasardarx wants to merge 1 commit into
ML4SCI:mainfrom
atharvasardarx:feature/improve-training-loop

Conversation

@atharvasardarx

Copy link
Copy Markdown

This pull request resolves an import blocking SyntaxError in trainer.py caused by invalid nested double quotes within an f-string strftime() evaluation. Resolving this syntax bug unblocks custom module imports across the repository, allowing demo notebooks and model training workflows to execute as expected.

In addition to the engine fix, the default device allocation in 01_LorentzParT_demo.ipynb has been updated to evaluate torch.backends.mps.is_available(), enabling native Apple Silicon GPU acceleration alongside standard CUDA and CPU fallbacks.

All modified files have been verified locally to confirm that module imports execute cleanly without syntax errors and notebook cells run smoothly on MPS enabled environments.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant