Skip to content

Add MO visualization widget's occupations and entropies to qpe_stretched_n2 notebook#550

Open
v-agamshayit wants to merge 9 commits into
mainfrom
agamshayit/mo_viewer_notebook
Open

Add MO visualization widget's occupations and entropies to qpe_stretched_n2 notebook#550
v-agamshayit wants to merge 9 commits into
mainfrom
agamshayit/mo_viewer_notebook

Conversation

@v-agamshayit

Copy link
Copy Markdown
Contributor

No description provided.

@v-agamshayit v-agamshayit marked this pull request as ready for review June 26, 2026 18:52
Copilot AI review requested due to automatic review settings June 26, 2026 18:52

Copilot AI left a comment

Copy link
Copy Markdown
Contributor

Choose a reason for hiding this comment

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

Pull request overview

This PR enhances the qpe_stretched_n2 example by augmenting the molecular-orbital visualization to display correlation diagnostics (natural occupations and single-orbital entropies) derived from the CASCI wavefunction, using the MoleculeViewer overlay metadata mechanism.

Changes:

  • Added a visualization utility to generate cube data augmented with active-space occupations and single-orbital entropies from a Wavefunction.
  • Updated the qpe_stretched_n2 notebook to request 1- and 2-RDMs from CASCI and to use the new utility for the visualization step.

Reviewed changes

Copilot reviewed 2 out of 2 changed files in this pull request and generated 1 comment.

File Description
examples/utils/viz_utils.py Adds generate_cube_data_with_correlation_info to build MoleculeViewer cube metadata (occupation/entropy) from a CASCI wavefunction.
examples/qpe_stretched_n2.ipynb Requests RDMs for CASCI and updates the visualization cell to overlay per-orbital occupations and entropies.

💡 Add Copilot custom instructions for smarter, more guided reviews. Learn how to get started.

Comment thread examples/utils/viz_utils.py Outdated
Co-authored-by: Copilot Autofix powered by AI <175728472+Copilot@users.noreply.github.com>
Copilot AI review requested due to automatic review settings June 26, 2026 23:13

Copilot AI left a comment

Copy link
Copy Markdown
Contributor

Choose a reason for hiding this comment

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

Pull request overview

Copilot reviewed 2 out of 2 changed files in this pull request and generated 2 comments.

Comment on lines +128 to +132
if indices is None and not orbitals.has_active_space():
raise ValueError(
"No active space is defined on the wavefunction's orbitals. "
"Set an active space on the Orbitals, or pass indices explicitly."
)
Comment on lines +147 to +148
occ_alpha, occ_beta = wavefunction.get_active_orbital_occupations()
mo_to_pos = {mo_idx: pos for pos, mo_idx in enumerate(active_indices)}
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.

MolecularViewer Widget could be improved by including additional information in the visualization

3 participants