luna-quantum 1.0.4rc1__cp311-cp311-macosx_11_0_arm64.whl → 1.0.4rc3__cp311-cp311-macosx_11_0_arm64.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=FQD2wNSk7QdKfwuvlrA52DGgdDYb2Xq1PtLwlu6bv-w,104
-luna_quantum-1.0.4rc1.dist-info/licenses/LICENSE,sha256=psuoW8kuDP96RQsdhzwOqi6fyWv0ct8CR6Jr7He_P_k,10173
-luna_quantum-1.0.4rc1.dist-info/licenses/NOTICE,sha256=noPOS8eDj5XoyRO8ZrCxIOh5fSjk0RildIrrqxQlepY,588
+luna_quantum-1.0.4rc3.dist-info/METADATA,sha256=0SVGFwI5zmYCuJShMOP6PA6cnWR6xJLRm4uBzGYHEy4,1585
+luna_quantum-1.0.4rc3.dist-info/WHEEL,sha256=FQD2wNSk7QdKfwuvlrA52DGgdDYb2Xq1PtLwlu6bv-w,104
+luna_quantum-1.0.4rc3.dist-info/licenses/LICENSE,sha256=psuoW8kuDP96RQsdhzwOqi6fyWv0ct8CR6Jr7He_P_k,10173
+luna_quantum-1.0.4rc3.dist-info/licenses/NOTICE,sha256=noPOS8eDj5XoyRO8ZrCxIOh5fSjk0RildIrrqxQlepY,588
 luna_quantum/__init__.py,sha256=RV92O-Th8ktm0S9kcpfRfECkeeORP-tbDKfpa4u1CiY,3223
 luna_quantum/__init__.pyi,sha256=1Joa1farT1Ey22yWYPMNPgQrGHIKL8bycrdPUgtQqw8,1661
-luna_quantum/_core.cpython-311-darwin.so,sha256=X39_wyMzgLcMpDm71yrgVL5p8NR0mh0neIRglKaXPBM,5116576
+luna_quantum/_core.cpython-311-darwin.so,sha256=wSDLZGWMaDePnj3_qoc77SBtWuY6MJVRU2UENyox-9c,5116576
 luna_quantum/_core.pyi,sha256=JnKcoRh_ZUjeWZRZl_K9usnPYHSaIzbYPuXnevjtoYY,113649
 luna_quantum/algorithms/__init__.py,sha256=IX9ZpzY3Do3mTgKqto5vAWwdYrZrM-RemYSf64yxefg,69
 luna_quantum/aqm_overwrites/__init__.py,sha256=rteObr5JHDnTnRime0Euq9Qy2iDIp6VMpFNHTGVNBe0,46
@@ -81,6 +81,7 @@ luna_quantum/client/schemas/wrappers/datetime_wrapper.py,sha256=8K2jexTarxGCgD1u
 luna_quantum/client/utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 luna_quantum/client/utils/qpu_token_utils.py,sha256=xeWIKcBwXGSQ0INfUY3NMTu6H90T03jkT85yHs2QXmk,4768
 luna_quantum/config.py,sha256=xx7DnjAvOeLUMlXvbXqQ4_EBAy53bjPYU1-QQO6I2lY,217
+luna_quantum/decorators.py,sha256=g4qVoWW_xkHM7QchbDDwPKmZkATKRA_7iJNzps8W1Eo,6837
 luna_quantum/errors.py,sha256=yjeXB3i07LlTuuJ18IKgwsC5wmtJElN8cDwlg3JkcEs,1424
 luna_quantum/errors.pyi,sha256=Hev30Htt63MgcESZaVOj645PsINfTYS-XGU8gklYqFM,9297
 luna_quantum/exceptions/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -244,7 +245,7 @@ luna_quantum/solve/usecases/solve_job_delete_usecase.py,sha256=hUOLeFLU9NCggWMGu
 luna_quantum/solve/usecases/solve_job_fetch_updates_usecase.py,sha256=W-Z6Q7IuyodjZmFKEsJGaa9V2fnGMPWj9FRN9gEdPNc,1686
 luna_quantum/solve/usecases/solve_job_get_result_usecase.py,sha256=dk6LSjInOfH5d7LlbUmawnqytKig_vFTCKTDK5KlDlI,3541
 luna_quantum/transformations.py,sha256=AZtGBaJ0PTWsr4mpONoJq5BpNOXPcM85CnWDhPgXx_I,902
-luna_quantum/transformations.pyi,sha256=-SJlobVvKUpJS0ZJOSod5iEdICAlSC8_XLtU--VfP4I,10775
+luna_quantum/transformations.pyi,sha256=vqx3I1hxKpQivG0WQ_60-Isocme_DD0veVAUO7xycuk,10768
 luna_quantum/translator.py,sha256=xi5eiIRNv8ATz69GYpUY5kbEIUmQEtK6_dF5t2Mwpec,1134
 luna_quantum/translator.pyi,sha256=edLGrIFl6do9LaCH4znjnOD8Zb3va-9nBGq-H3XQHso,26478
 luna_quantum/util/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -255,5 +256,5 @@ luna_quantum/util/log_utils.py,sha256=8WK0ivN3Dq3QntIYu-eGRBrGyiBoFnZkRiX_GCc-IH
 luna_quantum/util/pretty_base.py,sha256=Vv3dYpMsydOPYFOm-0lCCuJIe-62oYkI64Zy_bahl3s,2115
 luna_quantum/util/pydantic_utils.py,sha256=nhl_SdLJVAizrtLVHvnbco84g8CdBVdVxN_jlXiv82w,1263
 luna_quantum/utils.py,sha256=pBOkGXNJXlOzxAwTJv8nCj32Q6WNeh3t6Ka3lmTgy9c,134
-luna_quantum/utils.pyi,sha256=afjDJhGiM9KmN2zqVkd6agP8JAsbwUxJBW9tEyiFrdY,1941
-luna_quantum-1.0.4rc1.dist-info/RECORD,,
+luna_quantum/utils.pyi,sha256=yHHPluEJArUltZ2jJ9bPtTugj59E9TOTmYdyH35EHtU,1934
+luna_quantum-1.0.4rc3.dist-info/RECORD,,