From 0ecd822b0490747a158552945d613cb3a681a695 Mon Sep 17 00:00:00 2001 From: Parth Sharma Date: Tue, 13 May 2025 18:42:22 +0530 Subject: [PATCH] Fix utils_tests --- tests/utils_tests/models.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/tests/utils_tests/models.py b/tests/utils_tests/models.py index 866a37debc44..ff8ad1031c7d 100644 --- a/tests/utils_tests/models.py +++ b/tests/utils_tests/models.py @@ -1,4 +1,6 @@ from django.db import models +from django_singlestore.schema import ModelStorageManager + class Category(models.Model): @@ -7,3 +9,4 @@ class Category(models.Model): class CategoryInfo(models.Model): category = models.OneToOneField(Category, models.CASCADE) + objects = ModelStorageManager("REFERENCE")