Skip to content

Commit 84ec89c

Browse files
Update generative-proof-of-concept-CPU-preprocessing-in-memory.py
Remove nesting ...
1 parent aa9281d commit 84ec89c

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

generative-proof-of-concept-CPU-preprocessing-in-memory.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1357,8 +1357,8 @@ def __next__(self):
13571357
# Pop and return one sample
13581358
# input_sample = [self.data.pop(0)] # Nested as per model input spec
13591359
# label_sample = [self.labels.pop(0)] # Nested as per model output spec
1360-
input_sample = [self.data.pop(0)]
1361-
label_sample = [self.labels.pop(0)]
1360+
input_sample = self.data.pop(0)
1361+
label_sample = self.labels.pop(0)
13621362

13631363
return (input_sample, label_sample)
13641364

0 commit comments

Comments
 (0)