Skip to content

Commit 630eff6

Browse files
committed
Revert dad8b00
1 parent 8cf0432 commit 630eff6

File tree

1 file changed

+2
-5
lines changed

1 file changed

+2
-5
lines changed

plasma/primitives/shots.py

Lines changed: 2 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -132,11 +132,8 @@ def sublists(self,num,do_shuffle=True,equal_size=False):
132132
self.shuffle()
133133
for i in range(0,len(self),num):
134134
subl = self.shots[i:i+num]
135-
#take into account the case of small shot lists < num (e.g. if used for testing)
136-
#do not expand the shot list with duplicate events, rather add random patches later on
137-
if len(self) >= num:
138-
while equal_size and len(subl) < num:
139-
subl.append(rnd.choice(self.shots))
135+
while equal_size and len(subl) < num:
136+
subl.append(rnd.choice(self.shots))
140137
lists.append(subl)
141138
return [ShotList(l) for l in lists]
142139

0 commit comments

Comments
 (0)