Merge pull request #21 from edx/jibsheet/dont-use-using

Jibsheet/dont use using

milestones/migrations/0002_data__seed_relationship_types.py

@@ -9,9 +9,8 @@ from milestones.data import fetch_milestone_relationship_types
 def seed_relationship_types(apps, schema_editor):
     """Seed the relationship types."""
     MilestoneRelationshipType = apps.get_model("milestones", "MilestoneRelationshipType")
-    db_alias = schema_editor.connection.alias
     for name in fetch_milestone_relationship_types().values():
-        MilestoneRelationshipType.objects.using(db_alias).get_or_create(
+        MilestoneRelationshipType.objects.get_or_create(
             name=name,
             description='Autogenerated milestone relationship type "{}"'.format(name),
         )
@@ -20,9 +19,8 @@ def seed_relationship_types(apps, schema_editor):
 def delete_relationship_types(apps, schema_editor):
     """Clean up any relationships we made."""
     MilestoneRelationshipType = apps.get_model("milestones", "MilestoneRelationshipType")
-    db_alias = schema_editor.connection.alias
     for name in fetch_milestone_relationship_types().values():
-        MilestoneRelationshipType.objects.using(db_alias).filter(name=name).delete()
+        MilestoneRelationshipType.objects.filter(name=name).delete()
 
 
 class Migration(migrations.Migration):

setup.py

@@ -4,7 +4,7 @@ from setuptools import setup, find_packages
 
 setup(
     name='edx-milestones',
-    version='0.1.7',
+    version='0.1.8',
     description='Significant events module for Open edX',
     long_description=open('README.md').read(),
     author='edX',