Merge 4b570af into c108169

weni-ai · Oct 24, 2018 · a9200c7 · a9200c7
2 parents c108169 + 4b570af
commit a9200c7
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 0 deletions.
diff --git a/bothub/common/models.py b/bothub/common/models.py
@@ -481,6 +481,9 @@ def ready_for_train(self):
            not self.deleted.exists():
             return False
 
+        if self.examples.count() == 0:
+            return False
+
         return len(self.requirements_to_train) is 0
 
     @property

diff --git a/bothub/common/tests.py b/bothub/common/tests.py
@@ -871,6 +871,15 @@ def test_entity_dont_have_min_examples(self):
             entity='hi')
         self.assertTrue(self.repository.current_update().ready_for_train)
 
+    def test_no_examples(self):
+        example = RepositoryExample.objects.create(
+            repository_update=self.repository.current_update(),
+            text='hi',
+            intent='greet')
+        self.repository.current_update().start_training(self.owner)
+        example.delete()
+        self.assertFalse(self.repository.current_update().ready_for_train)
+
 
 class RequestRepositoryAuthorizationTestCase(TestCase):
     def setUp(self):