From 34e1711081cd14abc6775e7aa31a7490cb86d722 Mon Sep 17 00:00:00 2001 From: "Alinson S. Xavier" Date: Tue, 19 Jan 2021 22:02:17 -0600 Subject: [PATCH] Remove incorrect import --- miplearn/components/tests/test_composite.py | 4 +--- 1 file changed, 1 insertion(+), 3 deletions(-) diff --git a/miplearn/components/tests/test_composite.py b/miplearn/components/tests/test_composite.py index 16d2ac2..cec70d0 100644 --- a/miplearn/components/tests/test_composite.py +++ b/miplearn/components/tests/test_composite.py @@ -4,8 +4,6 @@ from unittest.mock import Mock, call -from nltk import Model - from miplearn import Component, LearningSolver, Instance from miplearn.components.composite import CompositeComponent @@ -14,7 +12,7 @@ def test_composite(): solver, instance, model = ( Mock(spec=LearningSolver), Mock(spec=Instance), - Mock(spec=Model), + Mock(), ) c1 = Mock(spec=Component)