Skip to content

Commit

Permalink
Merge 96baea3 into c108169
Browse files Browse the repository at this point in the history
  • Loading branch information
Douglas Paz committed Oct 25, 2018
2 parents c108169 + 96baea3 commit 332a243
Show file tree
Hide file tree
Showing 2 changed files with 19 additions and 2 deletions.
5 changes: 4 additions & 1 deletion bothub/common/models.py
Original file line number Diff line number Diff line change
Expand Up @@ -460,6 +460,9 @@ def ready_for_train(self):
if self.training_started_at:
return False

if len(self.requirements_to_train) > 0:
return False

previous_update = self.repository.updates.filter(
language=self.language,
by__isnull=False,
Expand All @@ -481,7 +484,7 @@ def ready_for_train(self):
not self.deleted.exists():
return False

return len(self.requirements_to_train) is 0
return True

@property
def intents(self):
Expand Down
16 changes: 15 additions & 1 deletion bothub/common/tests.py
Original file line number Diff line number Diff line change
Expand Up @@ -767,7 +767,8 @@ def setUp(self):
owner=self.owner,
name='Test',
slug='test',
language=languages.LANGUAGE_EN)
language=languages.LANGUAGE_EN,
use_language_model_featurizer=False)

def test_be_true(self):
RepositoryExample.objects.create(
Expand Down Expand Up @@ -871,6 +872,19 @@ def test_entity_dont_have_min_examples(self):
entity='hi')
self.assertTrue(self.repository.current_update().ready_for_train)

def test_settings_change_exists_requirements(self):
self.repository.current_update().start_training(self.owner)
self.repository.use_language_model_featurizer = True
self.repository.save()
RepositoryExample.objects.create(
repository_update=self.repository.current_update(),
text='hello',
intent='greet')
self.assertEqual(
len(self.repository.current_update().requirements_to_train),
1)
self.assertFalse(self.repository.current_update().ready_for_train)


class RequestRepositoryAuthorizationTestCase(TestCase):
def setUp(self):
Expand Down

0 comments on commit 332a243

Please sign in to comment.