Skip to content

Add quantumnas module to import chain#318

Open
yurekami wants to merge 1 commit into
mit-han-lab:mainfrom
yurekami:fix/issue-250-super-layers-import
Open

Add quantumnas module to import chain#318
yurekami wants to merge 1 commit into
mit-han-lab:mainfrom
yurekami:fix/issue-250-super-layers-import

Conversation

@yurekami

Copy link
Copy Markdown

Summary

  • Create __init__.py for torchquantum/algorithm/quantumnas package
  • Export all classes from super_layers, super_utils, and prune_utils
  • Update algorithm/__init__.py to include quantumnas imports

This enables standard import patterns:

from torchquantum.algorithm import Super1QLayer
from torchquantum.algorithm.quantumnas import ArchSampler, get_named_sample_arch

Test plan

  • Verify init.py properly exports all classes
  • Run import tests to verify the import chain works

Fixes #250

🤖 Generated with Claude Code

- Create __init__.py for torchquantum/algorithm/quantumnas
- Export super_layers, super_utils, and prune_utils classes
- Update algorithm/__init__.py to include quantumnas imports

This enables standard import patterns like:
  from torchquantum.algorithm import Super1QLayer
  from torchquantum.algorithm.quantumnas import ArchSampler

Fixes mit-han-lab#250

🤖 Generated with [Claude Code](https://claude.com/claude-code)

Co-Authored-By: Claude Opus 4.5 <noreply@anthropic.com>
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.

Run example of QuantumNAS with error "no nodule namsed 'torchquantum.super_layers'"

1 participant