Remove 'to' from kwargs in CategoryM2MField and CategoryFKField. 'to'…

… is already specified, and causes errors when running unit tests
jazzband · May 8, 2011 · 29ab094 · 29ab094
1 parent 1100ad3
commit 29ab094
Showing 1 changed file with 9 additions and 3 deletions.
diff --git a/categories/fields.py b/categories/fields.py
@@ -1,12 +1,18 @@
 from django.db.models import ForeignKey, ManyToManyField
 
+from categories.models import Category
+
+
+
 class CategoryM2MField(ManyToManyField):
     def __init__(self, **kwargs):
-        from categories.models import Category
+        if 'to' in kwargs:
+            kwargs.pop('to')
         super(CategoryM2MField, self).__init__(to=Category, **kwargs)
 
 
 class CategoryFKField(ForeignKey):
     def __init__(self, **kwargs):
-        from categories.models import Category
-        super(CategoryFKField, self).__init__(to=Category, **kwargs)
+        if 'to' in kwargs:
+            kwargs.pop('to')
+        super(CategoryFKField, self).__init__(to=Category, **kwargs)