Fix failing pytests

README.md

@@ -1,7 +1,7 @@
 # Pyoframe
 
 [![codecov](https://codecov.io/gh/staadecker/pyoframe/graph/badge.svg?token=8258XESRYQ)](https://codecov.io/gh/staadecker/pyoframe)
-![Build](https://github.com/staadecker/pyoframe/actions/workflows/ci.yml/badge.svg)
+[![Build](https://github.com/staadecker/pyoframe/actions/workflows/ci.yml/badge.svg)](https://github.com/staadecker/pyoframe/actions/workflows/ci.yml)
 [![Docs](https://github.com/staadecker/pyoframe/actions/workflows/publish_doc.yml/badge.svg)](https://staadecker.github.io/pyoframe/reference/)
 
 A library to rapidly formulate large optimization models using Pandas or Polars dataframes.

src/pyoframe/__init__.py

@@ -4,12 +4,12 @@
 """
 
 from pyoframe.monkey_patch import patch_dataframe_libraries
-from pyoframe.constraints import sum, Set
+from pyoframe.constraints import sum, sum_by, Set
 from pyoframe.constants import Config
 from pyoframe.variables import Variable
 from pyoframe.model import Model
 from pyoframe.constants import VType
 
 patch_dataframe_libraries()
 
-__all__ = ["sum", "Variable", "Model", "Set", "VType", "Config"]
+__all__ = ["sum", "sum_by", "Variable", "Model", "Set", "VType", "Config"]

src/pyoframe/objective.py

@@ -6,7 +6,7 @@
 class Objective(Expression):
     r"""
     Examples:
-        >>> from pyoframe import Objective, Variable, Model, sum
+        >>> from pyoframe import Variable, Model, sum
         >>> m = Model()
         >>> m.a = Variable()
         >>> m.b = Variable({"dim1": [1, 2, 3]})