diff --git a/recommender/models/service.py b/recommender/models/service.py index b9579ec0..6e9b5ed7 100644 --- a/recommender/models/service.py +++ b/recommender/models/service.py @@ -44,6 +44,7 @@ class Service(MarketplaceDocument): one_hot_tensor = ListField(IntField(), blank=True) dense_tensor = ListField(FloatField(), blank=True) status = StringField() + pid = StringField() meta = { "indexes": [ diff --git a/tests/factories/marketplace/service.py b/tests/factories/marketplace/service.py index 0877e9f6..2e5fcc5e 100644 --- a/tests/factories/marketplace/service.py +++ b/tests/factories/marketplace/service.py @@ -78,3 +78,4 @@ class Meta: related_services = [] required_services = [] status = "published" + pid = "pid"