Skip to content

Commit

Permalink
Fix default LagrangianStore types
Browse files Browse the repository at this point in the history
  • Loading branch information
merajhashemi committed Mar 18, 2024
1 parent f8d0b48 commit 3abd004
Showing 1 changed file with 3 additions and 3 deletions.
6 changes: 3 additions & 3 deletions cooper/cmp.py
Original file line number Diff line number Diff line change
@@ -1,6 +1,6 @@
import abc
from collections.abc import Sequence
from dataclasses import dataclass
from dataclasses import dataclass, field
from typing import Optional

import torch
Expand All @@ -20,8 +20,8 @@ class LagrangianStore:

lagrangian: Optional[torch.Tensor] = None
dual_lagrangian: Optional[torch.Tensor] = None
primal_constraint_measurements: Optional[list[ConstraintMeasurement]] = None
dual_constraint_measurements: Optional[list[ConstraintMeasurement]] = None
primal_constraint_measurements: list[ConstraintMeasurement] = field(default_factory=list)
dual_constraint_measurements: list[ConstraintMeasurement] = field(default_factory=list)

def primal_backward(self) -> None:
"""Triggers backward calls to compute the gradient of the Lagrangian with
Expand Down

0 comments on commit 3abd004

Please sign in to comment.