Skip to content

Commit

Permalink
Merge 92a4ef1 into a616534
Browse files Browse the repository at this point in the history
  • Loading branch information
johncordeiro committed Jan 26, 2019
2 parents a616534 + 92a4ef1 commit a8fd61c
Show file tree
Hide file tree
Showing 2 changed files with 38 additions and 24 deletions.
24 changes: 0 additions & 24 deletions bothub/common/migrations/0028_auto_20190121_1250.py
Original file line number Diff line number Diff line change
Expand Up @@ -3,21 +3,6 @@
from django.db import migrations, models


def populate_algorithm(apps, *args):
from bothub.common.models import Repository as R
Repository = apps.get_model('common', 'Repository')
for repository in Repository.objects.all():
repository.algorithm = R.ALGORITHM_NEURAL_NETWORK_EXTERNAL \
if repository.use_language_model_featurizer else \
R.ALGORITHM_NEURAL_NETWORK_INTERNAL
repository.save(update_fields=['algorithm'])
for update in repository.updates.all():
update.algorithm = R.ALGORITHM_NEURAL_NETWORK_EXTERNAL \
if update.use_language_model_featurizer else \
R.ALGORITHM_NEURAL_NETWORK_INTERNAL
update.save(update_fields=['algorithm'])


class Migration(migrations.Migration):

dependencies = [
Expand All @@ -35,13 +20,4 @@ class Migration(migrations.Migration):
name='algorithm',
field=models.CharField(choices=[('statistical_model', 'Statistical Model'), ('neural_network_internal', 'Neural Network with internal vocabulary'), ('neural_network_external', 'Neural Network with external vocabulary (BETA)')], default='statistical_model', max_length=24, verbose_name='algorithm'),
),
migrations.RunPython(populate_algorithm),
migrations.RemoveField(
model_name='repository',
name='use_language_model_featurizer',
),
migrations.RemoveField(
model_name='repositoryupdate',
name='use_language_model_featurizer',
),
]
38 changes: 38 additions & 0 deletions bothub/common/migrations/0029_auto_20190126_0247.py
Original file line number Diff line number Diff line change
@@ -0,0 +1,38 @@
# Generated by Django 2.1.5 on 2019-01-26 02:47

from django.db import migrations


def populate_algorithm(apps, *args):
from bothub.common.models import Repository as R
Repository = apps.get_model('common', 'Repository')
for repository in Repository.objects.all():
repository.algorithm = R.ALGORITHM_NEURAL_NETWORK_EXTERNAL \
if repository.use_language_model_featurizer else \
R.ALGORITHM_NEURAL_NETWORK_INTERNAL
repository.save(update_fields=['algorithm'])
for update in repository.updates.all():
update.algorithm = R.ALGORITHM_NEURAL_NETWORK_EXTERNAL \
if update.use_language_model_featurizer else \
R.ALGORITHM_NEURAL_NETWORK_INTERNAL
update.save(update_fields=['algorithm'])


class Migration(migrations.Migration):

dependencies = [
('common', '0028_auto_20190121_1250'),
]

operations = [
migrations.RunPython(populate_algorithm),
migrations.RemoveField(
model_name='repository',
name='use_language_model_featurizer',
),
migrations.RemoveField(
model_name='repositoryupdate',
name='use_language_model_featurizer',
),
]

0 comments on commit a8fd61c

Please sign in to comment.