Skip to content

Optimizer and Scheduler Type Hinting #186

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
arthurmccray merged 9 commits into from
Mar 5, 2026
Merged

Optimizer and Scheduler Type Hinting #186

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
9 commits
Select commit Hold shift + click to select a range
059903a
Only dev stuff
cedriclim1 Mar 4, 2026
28f50cd
adding docstrings to OptimizerParams and SchedulerParams and promotin…
arthurmccray Mar 4, 2026
825c50c
adding backwards compatibility for "type" rather than "name"
arthurmccray Mar 4, 2026
5cbcd88
left a print
arthurmccray Mar 4, 2026
1946c87
Added num_iters back in when calling .params
cedriclim1 Mar 5, 2026
a46a75a
fixing Exponential scheduler
arthurmccray Mar 5, 2026
4d9b9fa
Fixed bug in parse_dict None -> none
cedriclim1 Mar 5, 2026
80d2b05
Merge branch 'opt_sched_typehinting' of https://github.com/electronmi…
cedriclim1 Mar 5, 2026
342bae6
Pytests and changed some defaults in optimizer_mixin.py
cedriclim1 Mar 5, 2026
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
4 changes: 4 additions & 0 deletions src/quantem/core/ml/__init__.py
View file
Open in desktop
Original file line number Diff line number Diff line change
@@ -1,4 +1,8 @@
from quantem.core.ml.cnn import CNN2d as CNN2d, CNN3d as CNN3d
from quantem.core.ml.optimizer_mixin import (
OptimizerParams as OptimizerParams,
SchedulerParams as SchedulerParams,
)
from quantem.core.ml.inr import HSiren as HSiren
from quantem.core.ml.dense_nn import DenseNN as DenseNN
from quantem.core.ml.cnn_dense import CNNDense as CNNDense
Expand Down
Loading