Adds assert

cvxgrp · May 13, 2024 · df77d2d · df77d2d
1 parent a0a5dac
commit df77d2d
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/dsp/cone_transforms.py b/dsp/cone_transforms.py
@@ -76,6 +76,7 @@ def scalar_multiply(self, scalar: float) -> KRepresentation:
     def constant_repr(cls, value: float | int) -> KRepresentation:
         if not isinstance(value, (float, int)):
             assert value.size == 1
+            assert isinstance(value, np.ndarray)
             value = np.squeeze(value)
 
         return KRepresentation(