pepflow 0.1.0__py3-none-any.whl → 0.1.2__py3-none-any.whl

pepflow/point_test.py

@@ -1,3 +1,22 @@
+# Copyright: 2025 The PEPFlow Developers
+#
+# Licensed to the Apache Software Foundation (ASF) under one
+# or more contributor license agreements.  See the NOTICE file
+# distributed with this work for additional information
+# regarding copyright ownership.  The ASF licenses this file
+# to you under the Apache License, Version 2.0 (the
+# "License"); you may not use this file except in compliance
+# with the License.  You may obtain a copy of the License at
+#
+#   http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing,
+# software distributed under the License is distributed on an
+# "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
+# KIND, either express or implied.  See the License for the
+# specific language governing permissions and limitations
+# under the License.
+
 import time
 
 import numpy as np
@@ -30,6 +49,16 @@ def test_point_tag():
         p1 = point.Point(is_basis=True, eval_expression=None)
         p1.add_tag(tag="my_tag")
     assert p1.tags == ["my_tag"]
+    assert p1.tag == "my_tag"
+
+
+def test_point_repr():
+    pep_builder = pep.PEPBuilder()
+    with pep_builder.make_context("test"):
+        p1 = point.Point(is_basis=True)
+        print(p1)  # it should be fine without tag
+        p1.add_tag("my_tag")
+        assert str(p1) == "my_tag"
 
 
 def test_scalar_tag():
@@ -38,6 +67,16 @@ def test_scalar_tag():
         s1 = scalar.Scalar(is_basis=True, eval_expression=None)
         s1.add_tag(tag="my_tag")
     assert s1.tags == ["my_tag"]
+    assert s1.tag == "my_tag"
+
+
+def test_scalar_repr():
+    pep_builder = pep.PEPBuilder()
+    with pep_builder.make_context("test"):
+        s1 = scalar.Scalar(is_basis=True)
+        print(s1)  # it should be fine without tag
+        s1.add_tag("my_tag")
+        assert str(s1) == "my_tag"
 
 
 def test_point_in_a_list():
@@ -224,26 +263,33 @@ def test_expression_manager_eval_point_large_scale():
         for pp in ctx.points:
             pm.eval_point(pp)
 
-        assert (time.time() - t) < 0.1
+        assert (time.time() - t) < 0.5
 
 
 def test_function_generate_triplet():
     pep_builder = pep.PEPBuilder()
     with pep_builder.make_context("test") as ctx:
         f = fc.Function(is_basis=True, reuse_gradient=True)
+        f.add_tag("f")
         g = fc.Function(is_basis=True, reuse_gradient=True)
+        g.add_tag("g")
         h = 5 * f + 5 * g
+        h.add_tag("h")
 
         f1 = fc.Function(is_basis=True, reuse_gradient=False)
+        f1.add_tag("f1")
         g1 = fc.Function(is_basis=True, reuse_gradient=False)
+        g1.add_tag("g1")
         h1 = 5 * f1 + 5 * g1
+        h1.add_tag("h1")
 
         p1 = point.Point(is_basis=True)
-        _, func_value, grad = h.generate_triplet(p1)
-        _, func_value_1, grad_1 = h.generate_triplet(p1)
+        p1.add_tag("p1")
+        p1_triplet = h.generate_triplet(p1)
+        p1_triplet_1 = h.generate_triplet(p1)
 
-        _, func_value_2, grad_2 = h1.generate_triplet(p1)
-        _, func_value_3, grad_3 = h1.generate_triplet(p1)
+        p1_triplet_2 = h1.generate_triplet(p1)
+        p1_triplet_3 = h1.generate_triplet(p1)
 
         pm = exm.ExpressionManager(ctx)
 
@@ -252,31 +298,31 @@ def test_function_generate_triplet():
     )
 
     np.testing.assert_allclose(
-        pm.eval_point(grad).vector, np.array([0, 5, 5, 0, 0, 0, 0])
+        pm.eval_point(p1_triplet.gradient).vector, np.array([0, 5, 5, 0, 0, 0, 0])
     )
     np.testing.assert_allclose(
-        pm.eval_scalar(func_value).vector, np.array([5, 5, 0, 0])
+        pm.eval_scalar(p1_triplet.function_value).vector, np.array([5, 5, 0, 0])
     )
 
     np.testing.assert_allclose(
-        pm.eval_point(grad_1).vector, np.array([0, 5, 5, 0, 0, 0, 0])
+        pm.eval_point(p1_triplet_1.gradient).vector, np.array([0, 5, 5, 0, 0, 0, 0])
     )
     np.testing.assert_allclose(
-        pm.eval_scalar(func_value_1).vector, np.array([5, 5, 0, 0])
+        pm.eval_scalar(p1_triplet_1.function_value).vector, np.array([5, 5, 0, 0])
     )
 
     np.testing.assert_allclose(
-        pm.eval_point(grad_2).vector, np.array([0, 0, 0, 5, 5, 0, 0])
+        pm.eval_point(p1_triplet_2.gradient).vector, np.array([0, 0, 0, 5, 5, 0, 0])
     )
     np.testing.assert_allclose(
-        pm.eval_scalar(func_value_2).vector, np.array([0, 0, 5, 5])
+        pm.eval_scalar(p1_triplet_2.function_value).vector, np.array([0, 0, 5, 5])
     )
 
     np.testing.assert_allclose(
-        pm.eval_point(grad_3).vector, np.array([0, 0, 0, 0, 0, 5, 5])
+        pm.eval_point(p1_triplet_3.gradient).vector, np.array([0, 0, 0, 0, 0, 5, 5])
     )
     np.testing.assert_allclose(
-        pm.eval_scalar(func_value_3).vector, np.array([0, 0, 5, 5])
+        pm.eval_scalar(p1_triplet_3.function_value).vector, np.array([0, 0, 5, 5])
     )
 
 
@@ -284,28 +330,24 @@ def test_function_add_stationary_point():
     pep_builder = pep.PEPBuilder()
     with pep_builder.make_context("test") as ctx:
         f = fc.Function(is_basis=True, reuse_gradient=True)
-        x_opt = f.add_stationary_point()
+        f.add_tag("f")
+        x_opt = f.add_stationary_point("x_opt")
 
         pm = exm.ExpressionManager(ctx)
 
-    np.testing.assert_allclose(pm.eval_point(x_opt).vector, np.array([1, 0]))
-
-    np.testing.assert_allclose(
-        pm.eval_scalar(f.constraints[0].scalar).matrix, [[0, 0], [0, 1]]
-    )
-    np.testing.assert_allclose(pm.eval_scalar(f.constraints[0].scalar).vector, [0])
-    np.testing.assert_allclose(pm.eval_scalar(f.constraints[0].scalar).constant, 0)
-    assert f.constraints[0].comparator == utils.Comparator.EQ
+        np.testing.assert_allclose(pm.eval_point(x_opt).vector, np.array([1]))
 
 
 def test_smooth_interpolability_constraints():
     pep_builder = pep.PEPBuilder()
     with pep_builder.make_context("test") as ctx:
         f = fc.SmoothConvexFunction(L=1)
-        _ = f.add_stationary_point()
+        f.add_tag("f")
+        _ = f.add_stationary_point("x_opt")
 
         x_0 = point.Point(is_basis=True)
-        _, _, _ = f.generate_triplet(x_0)
+        x_0.add_tag("x_0")
+        _ = f.generate_triplet(x_0)
 
         all_interpolation_constraints = f.get_interpolation_constraints()
 
@@ -316,12 +358,7 @@ def test_smooth_interpolability_constraints():
     )
     np.testing.assert_allclose(
         pm.eval_scalar(all_interpolation_constraints[1].scalar).matrix,
-        [
-            [0.0, -0.5, 0.0, 0.0],
-            [-0.5, 0.5, 0.5, -0.5],
-            [0.0, 0.5, 0.0, 0.0],
-            [0.0, -0.5, 0.0, 0.5],
-        ],
+        [[0.0, 0.0, 0.0], [0.0, 0.0, 0.0], [0.0, 0.0, 0.5]],
     )
 
     np.testing.assert_allclose(

pepflow/scalar.py

@@ -1,3 +1,22 @@
+# Copyright: 2025 The PEPFlow Developers
+#
+# Licensed to the Apache Software Foundation (ASF) under one
+# or more contributor license agreements.  See the NOTICE file
+# distributed with this work for additional information
+# regarding copyright ownership.  The ASF licenses this file
+# to you under the Apache License, Version 2.0 (the
+# "License"); you may not use this file except in compliance
+# with the License.  You may obtain a copy of the License at
+#
+#   http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing,
+# software distributed under the License is distributed on an
+# "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
+# KIND, either express or implied.  See the License for the
+# specific language governing permissions and limitations
+# under the License.
+
 from __future__ import annotations
 
 import uuid
@@ -31,8 +50,8 @@ class EvalExpressionScalar:
 
 @attrs.frozen
 class EvaluatedScalar:
-    vector: np.array
-    matrix: np.array
+    vector: np.ndarray
+    matrix: np.ndarray
     constant: float
 
     def __add__(self, other):
@@ -140,9 +159,20 @@ class Scalar:
             raise RuntimeError("Did you forget to create a context?")
         pep_context.add_scalar(self)
 
+    @property
+    def tag(self):
+        if len(self.tags) == 0:
+            raise ValueError("Scalar should have a name.")
+        return self.tags[-1]
+
     def add_tag(self, tag: str):
         self.tags.append(tag)
 
+    def __repr__(self):
+        if self.tags:
+            return self.tag
+        return super().__repr__()
+
     def __add__(self, other):
         assert is_numerical_or_scalar(other)
         return Scalar(

pepflow/solver.py

@@ -1,13 +1,39 @@
+# Copyright: 2025 The PEPFlow Developers
+#
+# Licensed to the Apache Software Foundation (ASF) under one
+# or more contributor license agreements.  See the NOTICE file
+# distributed with this work for additional information
+# regarding copyright ownership.  The ASF licenses this file
+# to you under the Apache License, Version 2.0 (the
+# "License"); you may not use this file except in compliance
+# with the License.  You may obtain a copy of the License at
+#
+#   http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing,
+# software distributed under the License is distributed on an
+# "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
+# KIND, either express or implied.  See the License for the
+# specific language governing permissions and limitations
+# under the License.
+
 import warnings
 
 import cvxpy
 
+from pepflow import constants
 from pepflow import constraint as ctr
 from pepflow import expression_manager as exm
 from pepflow import pep_context as pc
 from pepflow import scalar as sc
 from pepflow import utils
 
+warnings.filterwarnings(
+    "ignore",
+    message=".*compressed sparse column.*",
+    category=UserWarning,
+)
+
 
 def evaled_scalar_to_cvx_express(
     eval_scalar: sc.EvaluatedScalar, vec_var: cvxpy.Variable, matrix_var: cvxpy.Variable
@@ -38,10 +64,9 @@ class DualVariableManager:
 
     def dual_value(self, name: str) -> float | None:
         if name not in self.named_constraints:
-            raise KeyError(f"Cannot find the constraint named {name}")
+            return None  # Is this good choice?
         dual_value = self.named_constraints[name].dual_value
         if dual_value is None:
-            warnings.warn("Did you forget to solve the problem first?")
             return None
         return dual_value
 
@@ -72,7 +97,7 @@ class CVXSolver:
             em.eval_scalar(scalar)
 
         self.dual_var_manager.clear()
-        self.dual_var_manager.add_constraint("PSD of Grammian Matrix", g_var >> 0)
+        self.dual_var_manager.add_constraint(constants.PSD_CONSTRAINT, g_var >> 0)
         for c in self.constraints:
             exp = evaled_scalar_to_cvx_express(em.eval_scalar(c.scalar), f_var, g_var)
             if c.comparator == utils.Comparator.GT:

pepflow/solver_test.py

@@ -1,3 +1,22 @@
+# Copyright: 2025 The PEPFlow Developers
+#
+# Licensed to the Apache Software Foundation (ASF) under one
+# or more contributor license agreements.  See the NOTICE file
+# distributed with this work for additional information
+# regarding copyright ownership.  The ASF licenses this file
+# to you under the Apache License, Version 2.0 (the
+# "License"); you may not use this file except in compliance
+# with the License.  You may obtain a copy of the License at
+#
+#   http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing,
+# software distributed under the License is distributed on an
+# "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
+# KIND, either express or implied.  See the License for the
+# specific language governing permissions and limitations
+# under the License.
+
 import numpy as np
 
 from pepflow import pep as pep

pepflow/utils.py

@@ -1,3 +1,22 @@
+# Copyright: 2025 The PEPFlow Developers
+#
+# Licensed to the Apache Software Foundation (ASF) under one
+# or more contributor license agreements.  See the NOTICE file
+# distributed with this work for additional information
+# regarding copyright ownership.  The ASF licenses this file
+# to you under the Apache License, Version 2.0 (the
+# "License"); you may not use this file except in compliance
+# with the License.  You may obtain a copy of the License at
+#
+#   http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing,
+# software distributed under the License is distributed on an
+# "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
+# KIND, either express or implied.  See the License for the
+# specific language governing permissions and limitations
+# under the License.
+
 from __future__ import annotations
 
 import enum
@@ -16,24 +35,6 @@ def SOP_self(v):
     return SOP(v, v)
 
 
-def merge_dict(dict1, dict2):
-    merged_dict = dict1.copy()
-    for key in dict2.keys():
-        if key in dict1.keys():
-            merged_dict[key] += dict2[key]
-        else:
-            merged_dict[key] = dict2[key]
-    return merged_dict
-
-
-def prune_dict(my_dict):
-    pruned_dict = dict()
-    for key in my_dict.keys():
-        if my_dict[key] != 0:
-            pruned_dict[key] = my_dict[key]
-    return pruned_dict
-
-
 class Op(enum.Enum):
     ADD = "add"
     SUB = "sub"

{pepflow-0.1.0.dist-info → pepflow-0.1.2.dist-info}/METADATA

@@ -1,15 +1,20 @@
 Metadata-Version: 2.4
 Name: pepflow
-Version: 0.1.0
-Summary: Add your description here
+Version: 0.1.2
+Summary: PEPFlow: A framework for Performance Estimation Problem (PEP) Workflow
 Requires-Python: >=3.9
 Description-Content-Type: text/markdown
 License-File: LICENSE
 Requires-Dist: attrs>=25.3.0
 Requires-Dist: cvxpy>=1.7.1
+Requires-Dist: dash>=3.2.0
+Requires-Dist: dash-bootstrap-components>=2.0.3
 Requires-Dist: isort>=6.0.1
 Requires-Dist: matplotlib>=3.9.4
+Requires-Dist: natsort>=8.4.0
 Requires-Dist: numpy>=2.0.2
+Requires-Dist: pandas>=2.3.1
+Requires-Dist: plotly>=6.3.0
 Requires-Dist: pytest>=8.4.1
 Requires-Dist: ruff>=0.12.4
 Requires-Dist: sympy>=1.14.0

pepflow-0.1.2.dist-info/RECORD

@@ -0,0 +1,22 @@
+pepflow/__init__.py,sha256=tLnOlZ1y_mIodRl5Fr5HMLPP5M7h_ad76s4in5OgiHE,1930
+pepflow/constants.py,sha256=t29CDRE8kw773zgKS0ZZCYGegwagaDDdLfpSaeDpK14,871
+pepflow/constraint.py,sha256=n-01dcQplvsXB7V4fceJBImbwSr-Wa9k9tE7ZcVmi3o,1153
+pepflow/expression_manager.py,sha256=64FbdMjcUrIqtbFZyJoxoDZ42j1bl1CeGxhJnKcGTD8,5222
+pepflow/function.py,sha256=yQcx78Ml6O34pVTcqYXxmO75l7IGdVvfwaY01lwsOHc,12566
+pepflow/function_test.py,sha256=947Jv0i7HtQ-LOi9-Hd7YsIGbX6BjnHsW3N4Q9HZE-A,4833
+pepflow/interactive_constraint.py,sha256=HMwAo80tJyCKwWKryxNGk1LAh-qlnQk5OeULLwKZUOQ,9826
+pepflow/pep.py,sha256=IKrYW6DBIekKOjGfL7rc6t1xU06VJEyL_gxj1sH_m_Q,5824
+pepflow/pep_context.py,sha256=IgrikL5Eqi0RfrFaVKn9lsodqtjj9A_RWFb82Qk-SYQ,4443
+pepflow/pep_context_test.py,sha256=1kFkmG56JVkmjaT9ic9byV-F9WOs9ZNQz5P1P5v5lLw,3092
+pepflow/pep_test.py,sha256=aWwP3CqXNJzQUhJqC5sNcsM83i_asJaZewpGNqC1G7M,2504
+pepflow/point.py,sha256=2iKmU6K_e3DmskCW9faQBxTaqPm3MUQPHTZvKVh3_yM,6711
+pepflow/point_test.py,sha256=jwbvjfwHP_9oY7KLp4UE_uSEbAVlDTXAQjr8d0z-eME,12602
+pepflow/scalar.py,sha256=vVA9qZjKIvM6RlZ4wN5MEFyG9eTFx7ItQaXtPY7ws7c,7846
+pepflow/solver.py,sha256=WzeN_IWNBs9IpE212jenhYMWFuuwH890h0vaFmJRM6I,4312
+pepflow/solver_test.py,sha256=BC37ggL_S8NDtnq8nPwRR2xItMCYMLs-n931u7f34YE,2414
+pepflow/utils.py,sha256=tYcpOOWAieNay5Ly7v7dRll69RPYCUGMEuW-Bwp2z68,1321
+pepflow-0.1.2.dist-info/licenses/LICENSE,sha256=na5oVXAps-5f1hLGG4SYnwFdavQeXgYUeN-E3MxOA_s,11361
+pepflow-0.1.2.dist-info/METADATA,sha256=tZyuaGvCHycaEKDj2RUQhhAguXM1F9JQYMpWbA-jRSc,1434
+pepflow-0.1.2.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+pepflow-0.1.2.dist-info/top_level.txt,sha256=0YEPCZQQa6yIAIwMumzDg4pj7AME8aXu2sXkuq8xM6M,8
+pepflow-0.1.2.dist-info/RECORD,,

pepflow-0.1.0.dist-info/RECORD

@@ -1,18 +0,0 @@
-pepflow/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-pepflow/constraint.py,sha256=-WSSpnzpso-Co4Fd-7B53mbTCB8pmgpPHgy3chZP0ws,324
-pepflow/expression_manager.py,sha256=xPo6i4tz_7e9c_2-iRen2i1-2JxLfR1b5HDqmaOtJ90,4393
-pepflow/function.py,sha256=hxbrvxUhBMRM8MdiNF9sP6oGaf-0swkuDW_Vp3Ogr90,6165
-pepflow/pep.py,sha256=I9MFXjtZnUBmnsqnixosZuV9_KHUthctXjxkCNuExcw,3688
-pepflow/pep_context.py,sha256=kh7iOAFWS-FMq_wiON1feQsbZLTPB4zARhe4mGR60Dk,707
-pepflow/pep_test.py,sha256=6gUhH8Gb2JOspoVwCFkf9NcoDRKwgs1BBFC367eL2hQ,1675
-pepflow/point.py,sha256=zKcWdNa24W__dTuappVWh134k12SuQF58XjDc8eZVBQ,5618
-pepflow/point_test.py,sha256=pxmxbCEmoF76RlrwQMl74Er6TdLt7_Fak4Wr2ymQn8o,11307
-pepflow/scalar.py,sha256=pSRvHztVKoIqUnVpIiwtcx3aP1wckFIzZTL5I3ZxXKg,6750
-pepflow/solver.py,sha256=v4eazPCsi8HhrHjTJ0Pf5OO4ABY3_Sns65UDyNZjOIc,3436
-pepflow/solver_test.py,sha256=dXdfCh91qp9itdcL3PfIAWuBxc7HdHo8XH3c9r5kWPk,1585
-pepflow/utils.py,sha256=v1LZbQXzwub55W05ou1LTiTI2MACTFw0FJt4K9lmFwU,918
-pepflow-0.1.0.dist-info/licenses/LICENSE,sha256=na5oVXAps-5f1hLGG4SYnwFdavQeXgYUeN-E3MxOA_s,11361
-pepflow-0.1.0.dist-info/METADATA,sha256=SEg5zgEa0fWxrQRA3E0uffnB_pxLmBJmemMSTNjlsnc,1226
-pepflow-0.1.0.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-pepflow-0.1.0.dist-info/top_level.txt,sha256=0YEPCZQQa6yIAIwMumzDg4pj7AME8aXu2sXkuq8xM6M,8
-pepflow-0.1.0.dist-info/RECORD,,