Skip to content

Added projective Wannierization utilities#3

Merged
Alphaharrius merged 5 commits into
mainfrom
dev/wannier
Apr 8, 2026
Merged

Added projective Wannierization utilities#3
Alphaharrius merged 5 commits into
mainfrom
dev/wannier

Conversation

@eyjafjallac
Copy link
Copy Markdown
Collaborator

@eyjafjallac eyjafjallac commented Apr 2, 2026

See #2

TODO

  • Momentum-space seed projection supported
  • Real-space seed projection supported
  • Test_case

- Implemented `projective_wannierization` for transforming target bands using seeding states in momentum space.
- Added `wannier_projection` for projective wannierization with real-space localized seed states, including Fourier transformation to momentum space.
- Included parameter validation and warnings for singular value decomposition (SVD) thresholds.
@eyjafjallac eyjafjallac added this to the v0.1.0 milestone Apr 2, 2026
@eyjafjallac eyjafjallac requested a review from Alphaharrius April 2, 2026 13:38
@eyjafjallac eyjafjallac self-assigned this Apr 2, 2026
@eyjafjallac eyjafjallac added the enhancement New feature or request label Apr 2, 2026
- Updated `pyproject.toml` to include mypy overrides for the `sympy` module to ignore missing imports.
- Modified `tox.ini` to change the mypy command to use `uv run`.
- Enhanced type annotations in `wannier.py` for better clarity and type safety.
- Added a new test suite in `test_wannier.py` to validate the functionality of the wannierization methods, including edge cases and expected behaviors.
@eyjafjallac eyjafjallac linked an issue Apr 2, 2026 that may be closed by this pull request
2 tasks
- Renamed `projective_wannierization` to `wannierize_k` and `wannier_projection` to `wannierize_r` for improved clarity.
- Updated `tox.ini` to modify the mypy command by removing the tests directory from the command.
- Added type stubs for the new function names in `wannier.pyi`.
- Adjusted tests in `test_wannier.py` to reflect the new function names and ensure proper validation.
- Introduced `projective_wannierization` to handle automatic seed-space dispatch for wannierization.
- Updated type stubs in `wannier.pyi` to include the new function.
- Added tests in `test_wannier.py` to validate the functionality and error handling of the new method, ensuring it matches existing wannierization functions.
Copy link
Copy Markdown
Owner

@Alphaharrius Alphaharrius left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

LGTM!

@Alphaharrius Alphaharrius merged commit 02e7f5d into main Apr 8, 2026
1 check passed
@Alphaharrius Alphaharrius deleted the dev/wannier branch April 8, 2026 15:02
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

enhancement New feature or request

Projects

None yet

Development

Successfully merging this pull request may close these issues.

Add Projective Wannierization

2 participants