luna-quantum 1.0.4rc1__cp311-cp311-win_amd64.whl → 1.0.4rc3__cp311-cp311-win_amd64.whl

luna_quantum/decorators.py

@@ -0,0 +1,247 @@
+"""Decorators."""
+
+from collections.abc import Callable
+from typing import Any, Generic, TypeVar, override
+
+from . import Model, Solution
+from .transformations import (
+    ActionType,
+    AnalysisCache,
+    AnalysisPass,
+    BasePass,
+    MetaAnalysisPass,
+    TransformationOutcome,
+    TransformationPass,
+)
+
+T = TypeVar("T")
+
+
+type AnalysisSignature[T] = Callable[[Model, AnalysisCache], T]
+
+type MetaAnalysisSignature[T] = Callable[[list[BasePass], AnalysisCache], T]
+
+type Outcome = (
+    TransformationOutcome | tuple[Model, ActionType] | tuple[Model, ActionType, Any]
+)
+type TransformationSignature = Callable[
+    [Model, AnalysisCache],
+    Outcome,
+]
+type BackwardsSignature = Callable[[Solution, AnalysisCache], Solution]
+
+
+def __identity_backwards(solution: Solution, _: AnalysisCache) -> Solution:
+    return solution
+
+
+class DynamicAnalysisPass(AnalysisPass, Generic[T]):
+    def __init__(
+        self,
+        name: str,
+        requires: list[str],
+        func: AnalysisSignature[T],
+    ) -> None:
+        self._name = name
+        self._requires = requires
+        self._func = func
+
+    @property
+    def name(self) -> str:
+        return self._name
+
+    @property
+    def requires(self) -> list[str]:
+        return self._requires
+
+    def __repr__(self) -> str:
+        return f'FunctionAnalysis(name="{self.name}")'
+
+    @override
+    def run(self, model: Model, cache: AnalysisCache) -> T:
+        return self._func(model, cache)
+
+    def __call__(self, model: Model, cache: AnalysisCache) -> T:
+        return self._func(model, cache)
+
+
+class DynamicMetaAnalysisPass(MetaAnalysisPass, Generic[T]):
+    def __init__(
+        self,
+        name: str,
+        requires: list[str],
+        func: MetaAnalysisSignature[T],
+    ) -> None:
+        self._name = name
+        self._requires = requires
+        self._func = func
+
+    @property
+    def name(self) -> str:
+        return self._name
+
+    @property
+    def requires(self) -> list[str]:
+        return self._requires
+
+    def __repr__(self) -> str:
+        return f'FunctionMetaAnalysis(name="{self.name}")'
+
+    @override
+    def run(self, passes: list[BasePass], cache: AnalysisCache) -> T:
+        return self._func(passes, cache)
+
+    def __call__(self, passes: list[BasePass], cache: AnalysisCache) -> T:
+        return self._func(passes, cache)
+
+
+class DynamicTransformationPass(TransformationPass):
+    def __init__(
+        self,
+        name: str,
+        requires: list[str],
+        invalidates: list[str],
+        func: TransformationSignature,
+        backwards: BackwardsSignature,
+    ) -> None:
+        self._name = name
+        self._requires = requires
+        self._invalidates = invalidates
+        self._func = func
+        self._backwards = backwards
+
+    @property
+    def name(self) -> str:
+        return self._name
+
+    @property
+    def requires(self) -> list[str]:
+        return self._requires
+
+    @override
+    def run(self, model: Model, cache: AnalysisCache) -> Outcome:
+        return self._func(model, cache)
+
+    @override
+    def backwards(self, solution: Solution, cache: AnalysisCache) -> Solution:
+        return self._backwards(solution, cache)
+
+    def __call__(self, model: Model, cache: AnalysisCache) -> Outcome:
+        return self._func(model, cache)
+
+    def __repr__(self) -> str:
+        return f'FunctionTransformation(name="{self.name}")'
+
+
+def analyse(
+    name: str | None = None, requires: list[str] | None = None
+) -> Callable[[AnalysisSignature[T]], DynamicAnalysisPass[T]]:
+    """Create an AnalysisPass instance from a function.
+
+    Parameters
+    ----------
+    name: str | None
+        The name of the analysis pass. If no name provided, uses the function name.
+    requires: list[str] | None
+        List of required analysis passes (defaults to empty list)
+
+    Returns
+    -------
+    Callable[[Callable[[Model, AnalysisCache], Any]], AnalysisPass]
+        An instance of a dynamically created AnalysisPass subclass
+    """
+    if requires is None:
+        requires = []
+
+    _T = TypeVar("_T")
+
+    def _decorator(
+        func: AnalysisSignature[_T],
+    ) -> DynamicAnalysisPass[_T]:
+        loc_name = name or func.__name__.replace("_", "-")
+
+        return DynamicAnalysisPass(name=loc_name, requires=requires, func=func)
+
+    return _decorator
+
+
+def meta_analyse(
+    name: str | None = None, requires: list[str] | None = None
+) -> Callable[[MetaAnalysisSignature[T]], DynamicMetaAnalysisPass[T]]:
+    """Create an MetaAnalysisPass instance from a function.
+
+    Parameters
+    ----------
+    name: str | None
+        The name of the analysis pass. If no name provided, uses the function name.
+    requires: list[str] | None
+        List of required analysis passes (defaults to empty list)
+
+    Returns
+    -------
+    Callable[[Callable[[list[BasePass], AnalysisCache], Any]], MetaAnalysisPass]
+        An instance of a dynamically created AnalysisPass subclass
+    """
+    if requires is None:
+        requires = []
+
+    _T = TypeVar("_T")
+
+    def _decorator(
+        func: MetaAnalysisSignature[_T],
+    ) -> DynamicMetaAnalysisPass[_T]:
+        loc_name = name or func.__name__.replace("_", "-")
+
+        return DynamicMetaAnalysisPass(name=loc_name, requires=requires, func=func)
+
+    return _decorator
+
+
+def transform(
+    name: str | None = None,
+    requires: list[str] | None = None,
+    invalidates: list[str] | None = None,
+    backwards: BackwardsSignature | None = None,
+) -> Callable[[TransformationSignature], DynamicTransformationPass]:
+    """Create an TransformationPass instance from a function.
+
+    Parameters
+    ----------
+    name: str | None = None
+        The name of the analysis pass. If no name provided, uses the function name.
+    requires: list[str] | None = None
+        List of required analysis passes (defaults to empty list)
+    invalidates: list[str] | None = None
+        List of analysis results to invalidate (defaults to empty list)
+    backwards: BackwardsSignature | None = None,
+        Solution backwards mapping function. If none provided, pass solution upstream
+        without modification.
+
+    Returns
+    -------
+    Callable[[TransformationSignature], DynamicTransformationPass]
+        An instance of a dynamically created TransformationPass subclass
+    """
+    if requires is None:
+        requires = []
+    if invalidates is None:
+        invalidates = []
+
+    if backwards is None:
+        backwards = __identity_backwards
+
+    def _decorator(func: TransformationSignature) -> DynamicTransformationPass:
+        loc_name = name or func.__name__.replace("_", "-")
+
+        return DynamicTransformationPass(
+            name=loc_name,
+            requires=requires,
+            invalidates=invalidates,
+            func=func,
+            backwards=backwards,
+        )
+
+    return _decorator
+
+
+__all__ = ["analyse", "transform"]

luna_quantum/transformations.pyi

@@ -3,7 +3,7 @@ from collections.abc import Callable
 from enum import Enum
 from typing import Any, Literal, overload
 
-from aqmodels import Model, Sense, Solution, Timing, Vtype
+from . import Model, Sense, Solution, Timing, Vtype
 
 class BasePass:
     @property

luna_quantum/utils.pyi

@@ -1,7 +1,7 @@
 from collections.abc import Generator, Iterable
 from typing import overload
 
-from aqmodels import Expression, Variable
+from . import Expression, Variable
 
 @overload
 def quicksum(iterable: Generator[Expression], /) -> Expression: ...

{luna_quantum-1.0.4rc1.dist-info → luna_quantum-1.0.4rc3.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: luna-quantum
-Version: 1.0.4rc1
+Version: 1.0.4rc3
 Classifier: Programming Language :: Python :: 3
 Classifier: License :: OSI Approved :: Apache Software License
 Classifier: Operating System :: OS Independent

{luna_quantum-1.0.4rc1.dist-info → luna_quantum-1.0.4rc3.dist-info}/RECORD

@@ -1,10 +1,10 @@
-luna_quantum-1.0.4rc1.dist-info/METADATA,sha256=Tg6Xtt1UIfDkUnxq6bBmPCP9MT7JYsa1rJsm66P61WE,1585
-luna_quantum-1.0.4rc1.dist-info/WHEEL,sha256=Ncll1xfKuKRSH9zs2vSEuHQoNZLcJWpzadJbiEni0qk,96
-luna_quantum-1.0.4rc1.dist-info/licenses/LICENSE,sha256=bR2Wj7Il7KNny38LiDGrASo12StUfpReF--OewXD5cw,10349
-luna_quantum-1.0.4rc1.dist-info/licenses/NOTICE,sha256=GHZYA5H4QFAhbhXliAi2SjWWXLjxl4hrHdEPyUbC0r0,601
+luna_quantum-1.0.4rc3.dist-info/METADATA,sha256=0SVGFwI5zmYCuJShMOP6PA6cnWR6xJLRm4uBzGYHEy4,1585
+luna_quantum-1.0.4rc3.dist-info/WHEEL,sha256=Ncll1xfKuKRSH9zs2vSEuHQoNZLcJWpzadJbiEni0qk,96
+luna_quantum-1.0.4rc3.dist-info/licenses/LICENSE,sha256=bR2Wj7Il7KNny38LiDGrASo12StUfpReF--OewXD5cw,10349
+luna_quantum-1.0.4rc3.dist-info/licenses/NOTICE,sha256=GHZYA5H4QFAhbhXliAi2SjWWXLjxl4hrHdEPyUbC0r0,601
 luna_quantum/__init__.py,sha256=9Dpk-wfL5fR_R74c-JsFKZmUy7OUfe4hj4ZXEJaPAt4,3342
 luna_quantum/__init__.pyi,sha256=9Md6njbzsErrktv0GlRGYSunQIrIyw5r0kEJx1yMpkw,1746
-luna_quantum/_core.cp311-win_amd64.pyd,sha256=5ui4_hE-v3eOzT7uiwd9yLWB9WvomiZzVIF2kAlQkKY,5993984
+luna_quantum/_core.cp311-win_amd64.pyd,sha256=DooaabF_Grh0VVyqOZvj-esP20tKy-1wdV6nsTQjGoE,5996032
 luna_quantum/_core.pyi,sha256=mezdmnUnyZqfYqFIKG4a1UtM3A6Y4RqSNqdybbH9q-E,117664
 luna_quantum/algorithms/__init__.py,sha256=EoTNO0FXXjPrhpYfN7q9c_nUhmVHk2nEZsbhaVOBh_g,70
 luna_quantum/aqm_overwrites/__init__.py,sha256=ieIVhfslOwrznbvwqu3vNzU_nFIHS1hyeUgIV097WdQ,49
@@ -81,6 +81,7 @@ luna_quantum/client/schemas/wrappers/datetime_wrapper.py,sha256=11KmaftPtKoafhcg
 luna_quantum/client/utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 luna_quantum/client/utils/qpu_token_utils.py,sha256=goq4nwmxKz0GTqLep01ydoopqHb336FPv6CXGGtR260,4915
 luna_quantum/config.py,sha256=a5UgYn6TIT2pFfdprvDy0RzyI4eAI_OFSdFSxpjQr2M,228
+luna_quantum/decorators.py,sha256=h9APzB-s3p_kCf0LfBPjG23hlswcWZfupK492DxUoJc,7084
 luna_quantum/errors.py,sha256=9qeMJY-hI1qlBqrWjaESVGx3ujEQuFdqONGgIl107kk,1458
 luna_quantum/errors.pyi,sha256=tf4cDmBtbE6Mjujl3B3nbT1zQzLM__iS9neFYA3chv4,9579
 luna_quantum/exceptions/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -244,7 +245,7 @@ luna_quantum/solve/usecases/solve_job_delete_usecase.py,sha256=a-lo9CoWjePo-aq7Y
 luna_quantum/solve/usecases/solve_job_fetch_updates_usecase.py,sha256=9vcs3x5uLmBjkqSF12IMHrPjrPSodm0AHI0qPVAruNE,1735
 luna_quantum/solve/usecases/solve_job_get_result_usecase.py,sha256=_62COwl-I1iGaX2lfrUnmZmPyKJZRNrJGtRVBiW1SYc,3636
 luna_quantum/transformations.py,sha256=A00BqmKpZz1Ixrb77a_ynpuqtB7w-AlVshg-QHp2UwM,920
-luna_quantum/transformations.pyi,sha256=GwrS0LBFoFuDJvtnjHWgrtZ-2mlEvWFQfSrYiaMkV40,11142
+luna_quantum/transformations.pyi,sha256=XjLxMhYGAeHABq19SHDBOTDEsOcHaBgTBQzlJ-5SxpQ,11135
 luna_quantum/translator.py,sha256=5RDTzFzd5sw3JzxFQsHdZQWhz2KDl03nzik5dUHebHI,1157
 luna_quantum/translator.pyi,sha256=RcePKY4Ztw-uv0Dfyx2DTw1TBMyem_SWts1MmzYZfn0,27347
 luna_quantum/util/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -255,5 +256,5 @@ luna_quantum/util/log_utils.py,sha256=0-ARoGkp7xyTvw1i-pRJoq1ylDKNC4eDoVY2DG05DS
 luna_quantum/util/pretty_base.py,sha256=rJy28OKfNdB1j5xdzOmFrCdbw-Gb-JJ5Dcz3NvOpIcQ,2182
 luna_quantum/util/pydantic_utils.py,sha256=KipyyVaajjFB-krdPl_j5BLogaiPqn0L8mrJuLwZtic,1301
 luna_quantum/utils.py,sha256=RlF0LFK0qXavL22BSjMuNcGBGJrEL8mde_rAPX66qhw,137
-luna_quantum/utils.pyi,sha256=JbBq1V2SpGy4IbwqCQQKoJUM17LiTD35Wv8Im1YY1pw,2008
-luna_quantum-1.0.4rc1.dist-info/RECORD,,
+luna_quantum/utils.pyi,sha256=TRV-THwZJdYDD5q3kn4W-p4dslim54xNJW0yFbXPFhs,2001
+luna_quantum-1.0.4rc3.dist-info/RECORD,,