diff --git a/thinc/tests/backends/test_ops.py b/thinc/tests/backends/test_ops.py index 2712f741a..df57d1d3a 100644 --- a/thinc/tests/backends/test_ops.py +++ b/thinc/tests/backends/test_ops.py @@ -15,7 +15,7 @@ MAX_EXAMPLES = 10 -VANILLA_OPS = Ops(numpy) +VANILLA_OPS = Ops(numpy) # type:ignore NUMPY_OPS = NumpyOps() BLIS_OPS = NumpyOps(use_blis=True) CPU_OPS = [NUMPY_OPS, VANILLA_OPS] @@ -365,7 +365,7 @@ def get_lstm_args(depth, dirs, nO, batch_size, nI, draw=None): if draw: params = draw(ndarrays_of_shape(n_params)) # For some reason this is crashing hypothesis? - #size_at_t = draw(ndarrays_of_shape(shape=(batch_size,), lo=1, dtype="int32")) + # size_at_t = draw(ndarrays_of_shape(shape=(batch_size,), lo=1, dtype="int32")) size_at_t = numpy.ones(shape=(batch_size,), dtype="int32") X = draw(ndarrays_of_shape((int(size_at_t.sum()), nI))) else: diff --git a/thinc/tests/test_util.py b/thinc/tests/test_util.py index 3736aaaba..69aac829a 100644 --- a/thinc/tests/test_util.py +++ b/thinc/tests/test_util.py @@ -13,13 +13,13 @@ (numpy.array(1), 0), (numpy.array([1, 2]), 3), ([numpy.zeros((1, 2)), numpy.zeros((1))], 2), - (Ragged(numpy.zeros((1, 2)), numpy.zeros(1)), 2), + (Ragged(numpy.zeros((1, 2)), numpy.zeros(1)), 2), # type:ignore ( Padded( - numpy.zeros((2, 1, 2)), - numpy.zeros(2), - numpy.array([1, 0]), - numpy.array([0, 1]), + numpy.zeros((2, 1, 2)), # type:ignore + numpy.zeros(2), # type:ignore + numpy.array([1, 0]), # type:ignore + numpy.array([0, 1]), # type:ignore ), 2, ),