Skip to content

Commit 46b6af5

Browse files
committed
Make tuple unpacking more explicit
1 parent 9521e56 commit 46b6af5

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

reptile/main.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -129,6 +129,7 @@ def REPTILE(
129129
epochs: int = EPOCHS,
130130
) -> Weights:
131131
"""Run one iteration of REPTILE."""
132+
132133
weights = [sgd(meta_weights, epochs) for _ in range(meta_batch_size)]
133134

134135
# TODO Implement custom optimizer that makes this work with builtin
@@ -187,7 +188,7 @@ def REPTILE(
187188
if PLOT:
188189

189190
ax.set_title(f'REPTILE after {iteration:n} iterations')
190-
curve, = ax.plot(
191+
(curve, ) = ax.plot(
191192
x_all.numpy(),
192193
model(Variable(x_all)).data.numpy(),
193194
label=f'Pred after {TEST_GRAD_STEPS} gradient steps.',

0 commit comments

Comments
 (0)