fix: failing test

ivyleavedtoadflax · ivyleavedtoadflax · commit 8e3a155648fe · 2020-04-11T21:56:41.000-04:00
diff --git a/tests/test_deep_reference_parser.py b/tests/test_deep_reference_parser.py
@@ -146,7 +146,7 @@ def test_DeepReferenceParser_predict(tmpdir, cfg):
         "And so is this".split(" "),
     ]
 
-    preds = drp.predict(examples, load_weights=True)
+    preds = drp.predict(examples, load_weights=True)[0]
 
     assert len(preds) == len(examples)
 

Original file line number	Diff line number	Diff line change
`@@ -146,7 +146,7 @@ def test_DeepReferenceParser_predict(tmpdir, cfg):`
`146`	`146`	`"And so is this".split(" "),`
`147`	`147`	`]`
`148`	`148`
`149`		`- preds = drp.predict(examples, load_weights=True)`
	`149`	`+ preds = drp.predict(examples, load_weights=True)[0]`
`150`	`150`
`151`	`151`	`assert len(preds) == len(examples)`
`152`	`152`