Fix ids in get_or_create

ou-astrophysics · Apr 12, 2024 · 239fa48 · 239fa48
1 parent c748c5b
commit 239fa48
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/starcatalogue/management/commands/create_all_zoo_subjects.py b/starcatalogue/management/commands/create_all_zoo_subjects.py
@@ -25,12 +25,12 @@ def handle(self, *args, **options):
 
             star = Star.objects.get_or_create(superwasp_id=superwasp_id)
             lightcurve = FoldedLightcurve.objects.get_or_create(
-                star=star,
+                star__id=star.id,
                 period_number=period_number,
             )
 
             ZooniverseSubject.objects.get_or_create(
-                zooniverse_id=subject_id, lightcurve=lightcurve
+                zooniverse_id=subject_id, lightcurve__id=lightcurve.id
             )
 
             imported_total += 1