pepflow 0.1.2__tar.gz → 0.1.3a1__tar.gz

{pepflow-0.1.2 → pepflow-0.1.3a1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: pepflow
-Version: 0.1.2
+Version: 0.1.3a1
 Summary: PEPFlow: A framework for Performance Estimation Problem (PEP) Workflow
 Requires-Python: >=3.9
 Description-Content-Type: text/markdown

{pepflow-0.1.2 → pepflow-0.1.3a1}/pepflow/function.py

@@ -347,3 +347,20 @@ class SmoothConvexFunction(Function):
                     self.smooth_convex_interpolability_constraints(i, j)
                 )
         return interpolation_constraints
+
+    def interpolate_ineq(
+        self, p1_tag: str, p2_tag: str, pep_context: pc.PEPContext | None = None
+    ) -> pt.Scalar:
+        """Generate the interpolation inequality scalar by tags."""
+        if pep_context is None:
+            pep_context = pc.get_current_context()
+        if pep_context is None:
+            raise RuntimeError("Did you forget to specify a context?")
+        # TODO: we definitely need a more robust tag system
+        x1 = pep_context.get_by_tag(p1_tag)
+        x2 = pep_context.get_by_tag(p2_tag)
+        f1 = pep_context.get_by_tag(f"{self.tag}({p1_tag})")
+        f2 = pep_context.get_by_tag(f"{self.tag}({p2_tag})")
+        g1 = pep_context.get_by_tag(f"gradient_{self.tag}({p1_tag})")
+        g2 = pep_context.get_by_tag(f"gradient_{self.tag}({p2_tag})")
+        return f2 - f1 + g2 * (x1 - x2) + 1 / 2 * (g1 - g2) ** 2

{pepflow-0.1.2 → pepflow-0.1.3a1}/pepflow/pep_context.py

@@ -52,6 +52,10 @@ class PEPContext:
         self.triplets: dict[Function, list[Triplet]] = defaultdict(list)
         self.opt_conditions: dict[Function, list[Constraint]] = defaultdict(list)
 
+    def set_as_current(self) -> PEPContext:
+        set_current_context(self)
+        return self
+
     def add_point(self, point: Point):
         self.points.append(point)
 
@@ -66,10 +70,10 @@ class PEPContext:
 
     def get_by_tag(self, tag: str) -> Point | Scalar:
         for p in self.points:
-            if p.tag == tag:
+            if tag in p.tags:
                 return p
         for s in self.scalars:
-            if s.tag == tag:
+            if tag in s.tags:
                 return s
         raise ValueError("Cannot find the point or scalar of given tag")
 

{pepflow-0.1.2 → pepflow-0.1.3a1}/pepflow/point.py

@@ -212,3 +212,14 @@ class Point:
         if not isinstance(other, Point):
             return NotImplemented
         return self.uid == other.uid
+
+    def eval(self, ctx: pc.PEPContext | None = None) -> np.ndarray:
+        from pepflow.expression_manager import ExpressionManager
+
+        # Note this can be inefficient.
+        if ctx is None:
+            ctx = pc.get_current_context()
+        if ctx is None:
+            raise RuntimeError("Did you forget to create a context?")
+        em = ExpressionManager(ctx)
+        return em.eval_point(self).vector

{pepflow-0.1.2 → pepflow-0.1.3a1}/pepflow/scalar.py

@@ -247,3 +247,14 @@ class Scalar:
 
     def eq(self, other, name: str) -> ctr.Constraint:
         return ctr.Constraint(self - other, comparator=utils.Comparator.EQ, name=name)
+
+    def eval(self, ctx: pc.PEPContext | None = None) -> EvaluatedScalar:
+        from pepflow.expression_manager import ExpressionManager
+
+        # Note this can be inefficient.
+        if ctx is None:
+            ctx = pc.get_current_context()
+        if ctx is None:
+            raise RuntimeError("Did you forget to create a context?")
+        em = ExpressionManager(ctx)
+        return em.eval_scalar(self)

{pepflow-0.1.2 → pepflow-0.1.3a1}/pepflow.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: pepflow
-Version: 0.1.2
+Version: 0.1.3a1
 Summary: PEPFlow: A framework for Performance Estimation Problem (PEP) Workflow
 Requires-Python: >=3.9
 Description-Content-Type: text/markdown

{pepflow-0.1.2 → pepflow-0.1.3a1}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "pepflow"
-version = "0.1.2"
+version = "0.1.3a1"
 description = "PEPFlow: A framework for Performance Estimation Problem (PEP) Workflow"
 readme = "README.md"
 requires-python = ">=3.9"