Add score field to Duplicate model

unicef · May 22, 2024 · ae004c0 · ae004c0
1 parent 3728173
commit ae004c0
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 2 deletions.
diff --git a/src/hope_dedup_engine/apps/api/migrations/0001_initial.py b/src/hope_dedup_engine/apps/api/migrations/0001_initial.py
@@ -1,4 +1,4 @@
-# Generated by Django 5.0.6 on 2024-05-21 07:23
+# Generated by Django 5.0.6 on 2024-05-22 13:24
 
 import django.db.models.deletion
 import uuid
@@ -63,9 +63,10 @@ class Migration(migrations.Migration):
             fields=[
                 ("id", models.BigAutoField(auto_created=True, primary_key=True, serialize=False, verbose_name="ID")),
                 ("first_reference_pk", models.IntegerField()),
-                ("second_reference_pk", models.IntegerField()),
                 ("first_filename", models.CharField(max_length=255)),
+                ("second_reference_pk", models.IntegerField()),
                 ("second_filename", models.CharField(max_length=255)),
+                ("score", models.FloatField()),
                 (
                     "deduplication_set",
                     models.ForeignKey(on_delete=django.db.models.deletion.CASCADE, to="api.deduplicationset"),

diff --git a/src/hope_dedup_engine/apps/api/models/deduplication.py b/src/hope_dedup_engine/apps/api/models/deduplication.py
@@ -50,3 +50,4 @@ class Duplicate(models.Model):
     first_filename = models.CharField(max_length=255)
     second_reference_pk = models.IntegerField()
     second_filename = models.CharField(max_length=255)
+    score = models.FloatField()
diff --git a/tests/extras/testutils/factories/api.py b/tests/extras/testutils/factories/api.py
@@ -36,6 +36,7 @@ class DuplicateFactory(DjangoModelFactory):
     first_reference_pk = fuzzy.FuzzyInteger(low=1)
     second_filename = fuzzy.FuzzyText()
     second_reference_pk = fuzzy.FuzzyInteger(low=1)
+    score = fuzzy.FuzzyFloat(low=0, high=1)
 
     class Meta:
         model = Duplicate