💄 Polish more

laminlabs · Oct 11, 2023 · e5132de · e5132de
1 parent c6eaf4e
commit e5132de
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 2 deletions.
diff --git a/lnschema_core/migrations/0022_migrate_to_integer_pks.py b/lnschema_core/migrations/0022_migrate_to_integer_pks.py
@@ -40,7 +40,13 @@ class Migration(migrations.Migration):
         ("lnschema_core", "0021_dataset_storage_alter_dataset_file"),
     ]
 
-    operations = []  # type: ignore
+    operations = [
+        migrations.AlterField(
+            model_name="dataset",
+            name="transform",
+            field=models.ForeignKey(default=None, null=True, on_delete=django.db.models.deletion.PROTECT, related_name="output_datasets", to="lnschema_core.transform"),
+        ),
+    ]
 
 
 # repurpose old primary key

diff --git a/lnschema_core/models.py b/lnschema_core/models.py
@@ -1906,7 +1906,7 @@ class Dataset(Registry, Data):
     """The feature sets measured in this dataset (see :class:`~lamindb.FeatureSet`)."""
     ulabels = models.ManyToManyField("ULabel", through="DatasetULabel", related_name="datasets")
     """ULabels sampled in the dataset (see :class:`~lamindb.Feature`)."""
-    transform = models.ForeignKey(Transform, PROTECT, related_name="datasets", null=True, default=None)
+    transform = models.ForeignKey(Transform, PROTECT, related_name="output_datasets", null=True, default=None)
     """:class:`~lamindb.Transform` whose run created the dataset."""
     run = models.ForeignKey(Run, PROTECT, related_name="output_datasets", null=True, default=None)
     """:class:`~lamindb.Run` that created the `file`."""