Revert ID change

ou-astrophysics · Apr 12, 2024 · 1e34f64 · 1e34f64
1 parent 4311c03
commit 1e34f64
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
@@ -26,15 +26,15 @@ def handle(self, *args, **options):
 
             star, _ = Star.objects.get_or_create(superwasp_id=superwasp_id)
             lightcurve, _ = FoldedLightcurve.objects.get_or_create(
-                star__id=star.id,
+                star=star,
                 period_number=period_number,
             )
             if lightcurve.period_length is None:
                 lightcurve.period_length = period_length
                 lightcurve.save()
 
             ZooniverseSubject.objects.get_or_create(
-                zooniverse_id=subject_id, lightcurve__id=lightcurve.id
+                zooniverse_id=subject_id, lightcurve=lightcurve
             )
 
             imported_total += 1