From 047f3d7d94a6ef9dec904a8a468497c9dcab7506 Mon Sep 17 00:00:00 2001 From: svlandeg Date: Fri, 15 May 2020 13:25:00 +0200 Subject: [PATCH] remove ops argument for Adam --- spacy/tests/parser/test_add_label.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/spacy/tests/parser/test_add_label.py b/spacy/tests/parser/test_add_label.py index fb43458ae..647c9720c 100644 --- a/spacy/tests/parser/test_add_label.py +++ b/spacy/tests/parser/test_add_label.py @@ -29,7 +29,7 @@ def _train_parser(parser): fix_random_seed(1) parser.add_label("left") parser.begin_training([], **parser.cfg) - sgd = Adam(0.001, ops=NumpyOps()) + sgd = Adam(0.001) for i in range(5): losses = {} @@ -42,7 +42,7 @@ def _train_parser(parser): def test_add_label(parser): parser = _train_parser(parser) parser.add_label("right") - sgd = Adam(0.001, ops=NumpyOps()) + sgd = Adam(0.001) for i in range(100): losses = {} doc = Doc(parser.vocab, words=["a", "b", "c", "d"])