Specified action if a negative number of samples requested

jeffrey-hokanson · Jun 8, 2020 · 92295aa · 92295aa
1 parent 0c561ae
commit 92295aa
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/psdr/domains/euclidean.py b/psdr/domains/euclidean.py
@@ -620,6 +620,11 @@ def sample(self, draw = 1):
 			If we are unable to find a point in the domain satisfying the constraints
 		"""
 		draw = int(draw)
+
+		# If request a non positive number of samples, simply return zero
+		if draw <= 0:
+			return np.zeros((0, len(self)))
+
 		x_sample = self._sample(draw = draw)
 		if draw == 1: 
 			x_sample = x_sample.flatten()