Skip to content

Commit

Permalink
Fix type hint
Browse files Browse the repository at this point in the history
  • Loading branch information
merajhashemi committed Mar 11, 2024
1 parent 3fe3803 commit 690a8ad
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion cooper/constraints/constraint_state.py
Original file line number Diff line number Diff line change
Expand Up @@ -87,7 +87,7 @@ def extract_violations(self, do_unsqueeze=True) -> tuple[torch.Tensor, torch.Ten

return violation, strict_violation

def extract_constraint_features(self) -> torch.Tensor:
def extract_constraint_features(self) -> tuple[torch.Tensor, torch.Tensor]:
"""Extracts the constraint features from the constraint state.
If strict constraint features are not provided, attempts to patch them with the
differentiable constraint features. Similarly, if differentiable constraint
Expand Down

0 comments on commit 690a8ad

Please sign in to comment.